lliu01 commited on
Commit
10f5a89
·
verified ·
1 Parent(s): 583d322
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -253,7 +253,7 @@ end\n
253
 
254
  if __name__ == "__main__":
255
  args = parse_args()
256
- tokenizer = AutoTokenizer.from_pretrained("lliu01/fortios")
257
  model = AutoModelForCausalLM.from_pretrained(
258
  "lliu01/fortios",
259
  torch_dtype=torch.bfloat16,
 
253
 
254
  if __name__ == "__main__":
255
  args = parse_args()
256
+ tokenizer = AutoTokenizer.from_pretrained("lliu01/fortios_one_config")
257
  model = AutoModelForCausalLM.from_pretrained(
258
  "lliu01/fortios",
259
  torch_dtype=torch.bfloat16,