Spaces:
Sleeping
Sleeping
Update main.py
Browse fileschanged the model name
main.py
CHANGED
@@ -3,7 +3,7 @@ from fastapi import FastAPI
|
|
3 |
from pydantic import BaseModel
|
4 |
|
5 |
|
6 |
-
llm = AutoModelForCausalLM.from_pretrained("zephyr-7b-beta.
|
7 |
model_type='mistral',
|
8 |
max_new_tokens = 1096,
|
9 |
threads = 3,
|
|
|
3 |
from pydantic import BaseModel
|
4 |
|
5 |
|
6 |
+
llm = AutoModelForCausalLM.from_pretrained("zephyr-7b-beta.Q8_0.gguf",
|
7 |
model_type='mistral',
|
8 |
max_new_tokens = 1096,
|
9 |
threads = 3,
|