staedi commited on
Commit
da71ea5
·
verified ·
1 Parent(s): 2172cba

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -53,7 +53,7 @@ def init_model(model:str='llama3.1:8b-instruct-q4_0'):
53
 
54
  llm_model = HuggingFaceInferenceAPI(
55
  # model='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/'
56
- model='Qwen/Qwen2.5-Coder-32B-Instruct'
57
  )
58
 
59
  return llm_model
 
53
 
54
  llm_model = HuggingFaceInferenceAPI(
55
  # model='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/'
56
+ model_name='Qwen/Qwen2.5-Coder-32B-Instruct'
57
  )
58
 
59
  return llm_model