asthara commited on
Commit
dfe1cc2
·
1 Parent(s): 51ad4bc

update model to use

Browse files
Files changed (1) hide show
  1. agent.py +1 -2
agent.py CHANGED
@@ -6,11 +6,10 @@ from smolagents import (
6
  )
7
  from prompts import prompt_templates
8
 
9
- #model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
10
  model = InferenceClientModel(
11
  max_tokens=2096,
12
  temperature=0.5,
13
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
14
  custom_role_conversions=None,
15
  )
16
 
 
6
  )
7
  from prompts import prompt_templates
8
 
 
9
  model = InferenceClientModel(
10
  max_tokens=2096,
11
  temperature=0.5,
12
+ model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
13
  custom_role_conversions=None,
14
  )
15