O S I H commited on
Commit
abfec2c
1 Parent(s): 70f98c5
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -22,10 +22,11 @@ token = os.environ["HF_TOKEN"]
22
 
23
  model = AutoModelForCausalLM.from_pretrained(
24
  "microsoft/Phi-3-small-8k-instruct",
25
- token=token,
 
26
  trust_remote_code=True,
27
  )
28
- tok = AutoTokenizer.from_pretrained("microsoft/Phi-3-small-8k-instruct", token=token)
29
  terminators = [
30
  tok.eos_token_id,
31
  ]
 
22
 
23
  model = AutoModelForCausalLM.from_pretrained(
24
  "microsoft/Phi-3-small-8k-instruct",
25
+ # token=token,
26
+ torch_dtype="auto",
27
  trust_remote_code=True,
28
  )
29
+ tok = AutoTokenizer.from_pretrained("microsoft/Phi-3-small-8k-instruct")#, token=token
30
  terminators = [
31
  tok.eos_token_id,
32
  ]