Revert to previous model
Browse files
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/
|
| 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,
|