jesusgj commited on
Commit
d1be4ad
·
1 Parent(s): bcf1ac6

Modified files

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -182,8 +182,8 @@ def initialize_agent():
182
  return None
183
 
184
  try:
185
- model = InferenceClientModel(model_id="meta-llama/Llama-3.1-70B-Instruct-Turbo", token=api_keys['together'], provider="together")
186
- logging.info("✅ Primary model (Llama 3.1 70B) loaded successfully")
187
  except Exception as e:
188
  logging.warning(f"⚠️ Failed to load primary model, falling back. Error: {e}")
189
  model = InferenceClientModel(model_id="Qwen/Qwen2.5-7B-Instruct", token=api_keys['together'], provider="together")
 
182
  return None
183
 
184
  try:
185
+ model = InferenceClientModel(model_id="deepseek-ai/DeepSeek-R1-0528", token=api_keys['together'], provider="together")
186
+ logging.info("✅ Primary model deepseek-ai/DeepSeek-R1-0528 loaded successfully")
187
  except Exception as e:
188
  logging.warning(f"⚠️ Failed to load primary model, falling back. Error: {e}")
189
  model = InferenceClientModel(model_id="Qwen/Qwen2.5-7B-Instruct", token=api_keys['together'], provider="together")