WilliamRabuel commited on
Commit
ac660b1
·
verified ·
1 Parent(s): b14fdc6

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +5 -5
models.py CHANGED
@@ -2,9 +2,9 @@ from smolagents import InferenceClientModel
2
  import os
3
 
4
  # Configuration des modèles
5
- ORCHESTRATOR_MODEL = "deepseek-ai/DeepSeek-R1-0528" # "moonshotai/Kimi-K2-Instruct"
6
- CODE_AGENT_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct" # "moonshotai/Kimi-K2-Instruct"
7
- VISION_MODEL = "Qwen/Qwen2.5-VL-32B-Instruct" # Qwen/Qwen2.5-VL-72B-Instruct
8
  REASONING_MODEL = "deepseek-ai/DeepSeek-R1-0528"
9
 
10
  class ModelManager:
@@ -35,7 +35,7 @@ class ModelManager:
35
  max_tokens=4096,
36
  temperature=0.0,
37
  timeout=240,
38
- provider="together-ai"
39
  )
40
 
41
  # Modèle de vision
@@ -45,7 +45,7 @@ class ModelManager:
45
  max_tokens=2048,
46
  temperature=0.1,
47
  timeout=240,
48
- provider="fireworks-ai"
49
  )
50
 
51
  # Modèle de raisonnement
 
2
  import os
3
 
4
  # Configuration des modèles
5
+ ORCHESTRATOR_MODEL = "Qwen/Qwen3-235B-A22B-Instruct-2507"
6
+ CODE_AGENT_MODEL = "Qwen/Qwen3-Coder-480B-A35B-Instruct"
7
+ VISION_MODEL = "Qwen/Qwen2.5-VL-72B-Instruct"
8
  REASONING_MODEL = "deepseek-ai/DeepSeek-R1-0528"
9
 
10
  class ModelManager:
 
35
  max_tokens=4096,
36
  temperature=0.0,
37
  timeout=240,
38
+ provider="novita"
39
  )
40
 
41
  # Modèle de vision
 
45
  max_tokens=2048,
46
  temperature=0.1,
47
  timeout=240,
48
+ provider="novita"
49
  )
50
 
51
  # Modèle de raisonnement