test-chat

Runtime error

lliu01 commited on Jul 25, 2024

Commit

10f5a89

verified ·

1 Parent(s): 583d322

Try new

Files changed (1) hide show

app.py CHANGED Viewed

@@ -253,7 +253,7 @@ end\n
 if __name__ == "__main__":
     args = parse_args()
-    tokenizer = AutoTokenizer.from_pretrained("lliu01/fortios")
     model = AutoModelForCausalLM.from_pretrained(
         "lliu01/fortios",
         torch_dtype=torch.bfloat16,

 if __name__ == "__main__":
     args = parse_args()
+    tokenizer = AutoTokenizer.from_pretrained("lliu01/fortios_one_config")
     model = AutoModelForCausalLM.from_pretrained(
         "lliu01/fortios",
         torch_dtype=torch.bfloat16,