Hugging Face's logo

Spaces:
rasyosef
/
phi-2-chat
Sleeping

App Files Community
rasyosef commited on
Commit
276c17b
·
verified ·
1 Parent(s): 8c7aab7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ checkpoint = "rasyosef/phi-2-instruct-v0.1"
14
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
15
  model = AutoModelForCausalLM.from_pretrained(
16
  checkpoint,
17
- torch_dtype=torch.float16,
18
  device_map=DEVICE
19
  )
20
 
 
14
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
15
  model = AutoModelForCausalLM.from_pretrained(
16
  checkpoint,
17
+ torch_dtype=torch.float32,
18
  device_map=DEVICE
19
  )
20