Update main.py
Browse files
main.py
CHANGED
|
@@ -30,8 +30,8 @@ from datetime import timedelta
|
|
| 30 |
@cl.step(type="tool")
|
| 31 |
async def LLMistral():
|
| 32 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
|
| 33 |
-
repo_id = "mistralai/Ministral-8B-Instruct-2410"
|
| 34 |
-
|
| 35 |
#repo_id = "nvidia/Mistral-NeMo-Minitron-8B-Base"
|
| 36 |
#repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
|
| 37 |
#repo_id = "allenai/OLMoE-1B-7B-0924-Instruct"
|
|
|
|
| 30 |
@cl.step(type="tool")
|
| 31 |
async def LLMistral():
|
| 32 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
|
| 33 |
+
#repo_id = "mistralai/Ministral-8B-Instruct-2410"
|
| 34 |
+
repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
| 35 |
#repo_id = "nvidia/Mistral-NeMo-Minitron-8B-Base"
|
| 36 |
#repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
|
| 37 |
#repo_id = "allenai/OLMoE-1B-7B-0924-Instruct"
|