w601sxs commited on
Commit
cf8fdc4
·
verified ·
1 Parent(s): a0a05de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ ref_model = AutoModelForCausalLM.from_pretrained("w601sxs/b1ade-1b-bf16", torch_
11
  ref_model = ref_model
12
  ref_model.eval()
13
 
14
- tokenizer = AutoTokenizer.from_pretrained("w601sxs/b1ade-1b-bf16")
15
 
16
 
17
  class KeywordsStoppingCriteria(StoppingCriteria):
 
11
  ref_model = ref_model
12
  ref_model.eval()
13
 
14
+ tokenizer = AutoTokenizer.from_pretrained("w601sxs/b1ade-1b-bf16")
15
 
16
 
17
  class KeywordsStoppingCriteria(StoppingCriteria):