Remove version tag from Hugging Face model repository in agent.py for consistency.
Browse files
agent.py
CHANGED
|
@@ -13,7 +13,7 @@ load_dotenv()
|
|
| 13 |
|
| 14 |
# Base Hugging Face LLM used by the chat wrapper
|
| 15 |
base_llm = HuggingFaceEndpoint(
|
| 16 |
-
repo_id="deepseek-ai/DeepSeek-R1-0528
|
| 17 |
# deepseek-ai/DeepSeek-OCR:novita
|
| 18 |
task="text-generation",
|
| 19 |
temperature=0.0,
|
|
|
|
| 13 |
|
| 14 |
# Base Hugging Face LLM used by the chat wrapper
|
| 15 |
base_llm = HuggingFaceEndpoint(
|
| 16 |
+
repo_id="deepseek-ai/DeepSeek-R1-0528",
|
| 17 |
# deepseek-ai/DeepSeek-OCR:novita
|
| 18 |
task="text-generation",
|
| 19 |
temperature=0.0,
|