test-chat

Runtime error

lliu01 commited on Jul 25, 2024

Commit

4315e30

verified ·

1 Parent(s): 10f5a89

Made a mistake

Files changed (1) hide show

app.py CHANGED Viewed

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

     args = parse_args()
     tokenizer = AutoTokenizer.from_pretrained("lliu01/fortios_one_config")
     model = AutoModelForCausalLM.from_pretrained(
+        "lliu01/fortios_one_config",
         torch_dtype=torch.bfloat16,
         low_cpu_mem_usage=True
     )