gdnartea commited on
Commit
7482cf5
1 Parent(s): f1bd0cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,9 +19,9 @@ canary_model.change_decoding_strategy(decode_cfg)
19
 
20
 
21
  # Load the text processing model and tokenizer
22
- proc_tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-128k-instruct")
23
  proc_model = AutoModelForCausalLM.from_pretrained(
24
- "microsoft/Phi-3-mini-128k-instruct",
25
  trust_remote_code=True,
26
  )
27
  proc_model.eval()
 
19
 
20
 
21
  # Load the text processing model and tokenizer
22
+ proc_tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct")
23
  proc_model = AutoModelForCausalLM.from_pretrained(
24
+ "microsoft/Phi-3-mini-4k-instruct",
25
  trust_remote_code=True,
26
  )
27
  proc_model.eval()