Coool2 commited on
Commit
83bad99
·
verified ·
1 Parent(s): b8b247b

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
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/deepseek-r1-0528-qwen3-8b"
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,