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,
|