Spaces:
Configuration error
Configuration error
oremaz
commited on
Commit
·
dd8b529
1
Parent(s):
3935bca
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -67,7 +67,7 @@ def get_max_memory_config(max_memory_per_gpu):
|
|
| 67 |
return max_memory
|
| 68 |
return None
|
| 69 |
|
| 70 |
-
model_id = "deepseek/
|
| 71 |
proj_llm = HuggingFaceLLM(
|
| 72 |
model_name=model_id,
|
| 73 |
tokenizer_name=model_id,
|
|
|
|
| 67 |
return max_memory
|
| 68 |
return None
|
| 69 |
|
| 70 |
+
model_id = "deepseek-ai/DeepSeek-R1-0528-Qwen3-8B"
|
| 71 |
proj_llm = HuggingFaceLLM(
|
| 72 |
model_name=model_id,
|
| 73 |
tokenizer_name=model_id,
|