Hugging Face's logo

Spaces:
truongghieu
/
spaceTest
Paused

App Files Community
truongghieu commited on
Commit
ed52491
•
1 Parent(s): 9d21281

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -15,7 +15,6 @@ bnb_config = BitsAndBytesConfig(
15
  tokenizer = AutoTokenizer.from_pretrained("truongghieu/deci-finetuned", trust_remote_code=True)
16
  # Load model in this way if use GPU
17
  model = AutoModelForCausalLM.from_pretrained("truongghieu/deci-finetuned", trust_remote_code=True, quantization_config=bnb_config)
18
- model.to(device)
19
  # model = AutoModelForCausalLM.from_pretrained("truongghieu/deci-finetuned", trust_remote_code=True)
20
  # Move the model to the GPU if available
21
 
 
15
  tokenizer = AutoTokenizer.from_pretrained("truongghieu/deci-finetuned", trust_remote_code=True)
16
  # Load model in this way if use GPU
17
  model = AutoModelForCausalLM.from_pretrained("truongghieu/deci-finetuned", trust_remote_code=True, quantization_config=bnb_config)
 
18
  # model = AutoModelForCausalLM.from_pretrained("truongghieu/deci-finetuned", trust_remote_code=True)
19
  # Move the model to the GPU if available
20