thuyentruong commited on
Commit
080cdf3
·
verified ·
1 Parent(s): e3092f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ model_path='microsoft/Phi-4-mini-instruct'
12
 
13
  model = AutoModelForCausalLM.from_pretrained(
14
  model_path,
15
- device_map="auto",
16
- torch_dtype="auto",
17
  trust_remote_code=True,
18
  )
19
  tokenizer = AutoTokenizer.from_pretrained(model_path)
 
12
 
13
  model = AutoModelForCausalLM.from_pretrained(
14
  model_path,
15
+ #device_map="auto",
16
+ #torch_dtype="auto",
17
  trust_remote_code=True,
18
  )
19
  tokenizer = AutoTokenizer.from_pretrained(model_path)