Spaces:
Running
Running
Update app.py
Browse files
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):
|