ama2aifusion commited on
Commit
e8003cf
·
verified ·
1 Parent(s): 50c4995

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -148,7 +148,8 @@ def build_graph(provider: str = "huggingface"):
148
  chat = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
149
  elif provider == "huggingface":
150
  # Add huggingface endpoint
151
- repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
 
152
  chat = ChatHuggingFace(
153
  #llm=HuggingFaceEndpoint(
154
  # endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
 
148
  chat = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
149
  elif provider == "huggingface":
150
  # Add huggingface endpoint
151
+ #repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
152
+ repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
153
  chat = ChatHuggingFace(
154
  #llm=HuggingFaceEndpoint(
155
  # endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",