File size: 366 Bytes
7a26e4a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from smolagents import CodeAgent, HfApiModel, FinalAnswerTool
# Basic inference model
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
custom_role_conversions=None,
)
# Code Agent
agent = CodeAgent(
model=model,
tools=[FinalAnswerTool()],
max_steps=2
)
|