Nioi commited on
Commit
3ad9743
·
1 Parent(s): 9981054

workaround until inference fix

Browse files
Files changed (1) hide show
  1. agent.py +13 -2
agent.py CHANGED
@@ -7,7 +7,7 @@ Facultative
7
  - arxiv_search
8
  '''
9
 
10
- from smolagents import CodeAgent, HfApiModel, tool, DuckDuckGoSearchTool
11
 
12
 
13
  @tool
@@ -92,4 +92,15 @@ def rounder(a:float, n:int) -> float:
92
  def get_agent() -> CodeAgent:
93
  search_tool = DuckDuckGoSearchTool()
94
 
95
- return CodeAgent(tools=[add, subtract, multiply, divide, modulus, rounder, search_tool], model=HfApiModel(bill_to="NioiOrg"))
 
 
 
 
 
 
 
 
 
 
 
 
7
  - arxiv_search
8
  '''
9
 
10
+ from smolagents import CodeAgent, HfApiModel, tool, DuckDuckGoSearchTool, MLXModel
11
 
12
 
13
  @tool
 
92
  def get_agent() -> CodeAgent:
93
  search_tool = DuckDuckGoSearchTool()
94
 
95
+ model = MLXModel(
96
+ "mlx-community/Qwen2.5-Coder-32B-Instruct-4bit",
97
+ {
98
+ "temperature": 0.7,
99
+ "top_k": 20,
100
+ "top_p": 0.8,
101
+ "min_p": 0.05,
102
+ "num_ctx": 32768,
103
+ },
104
+ )
105
+
106
+ return CodeAgent(tools=[add, subtract, multiply, divide, modulus, rounder, search_tool], model=model)