lliu01 commited on
Commit
00bcc1d
·
verified ·
1 Parent(s): 7763d5c

Revert to previous model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ def predict(message, history, system_prompt, temperature, max_tokens):
70
 
71
  if __name__ == "__main__":
72
  args = parse_args()
73
- tokenizer = AutoTokenizer.from_pretrained("lliu01/fortios_cli")
74
  model = AutoModelForCausalLM.from_pretrained(
75
  "lliu01/fortios",
76
  torch_dtype=torch.bfloat16,
 
70
 
71
  if __name__ == "__main__":
72
  args = parse_args()
73
+ tokenizer = AutoTokenizer.from_pretrained("lliu01/fortios")
74
  model = AutoModelForCausalLM.from_pretrained(
75
  "lliu01/fortios",
76
  torch_dtype=torch.bfloat16,