Try new
Browse files
app.py
CHANGED
@@ -253,7 +253,7 @@ end\n
|
|
253 |
|
254 |
if __name__ == "__main__":
|
255 |
args = parse_args()
|
256 |
-
tokenizer = AutoTokenizer.from_pretrained("lliu01/
|
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,
|