Update main.py
Browse files
main.py
CHANGED
@@ -85,7 +85,7 @@ async def start():
|
|
85 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
|
86 |
repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
87 |
|
88 |
-
|
89 |
repo_id=repo_id, max_new_tokens=8000, temperature=1.0, task="text2text-generation", streaming=True
|
90 |
)
|
91 |
|
|
|
85 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
|
86 |
repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
87 |
|
88 |
+
model = HuggingFaceEndpoint(
|
89 |
repo_id=repo_id, max_new_tokens=8000, temperature=1.0, task="text2text-generation", streaming=True
|
90 |
)
|
91 |
|