Hugging Face's logo

Spaces:
truongghieu
/
spaceTest
Paused

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -13,7 +13,6 @@ bnb_config = BitsAndBytesConfig(
13
 
14
 
15
  tokenizer = AutoTokenizer.from_pretrained("truongghieu/deci-finetuned", trust_remote_code=True)
16
- tokenizer.to(device)
17
  # Load model in this way if use GPU
18
  model = AutoModelForCausalLM.from_pretrained("truongghieu/deci-finetuned", trust_remote_code=True, quantization_config=bnb_config)
19
  model.to(device)
 
13
 
14
 
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)