ama2aifusion commited on
Commit
bba8f18
·
verified ·
1 Parent(s): 6457d2d

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -149,9 +149,9 @@ def build_graph(provider: str = "huggingface"):
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" -> it doesn't reply well
153
  #repo_id = "deepseek-ai/DeepSeek-Coder-V2-Instruct" -> it doesn't work (error on StopIteration)
154
- repo_id = "meta-llama/CodeLlama-34b-Instruct-hf"
155
  chat = ChatHuggingFace(
156
  #llm=HuggingFaceEndpoint(
157
  # endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
 
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" # -> it doesn't reply well
153
  #repo_id = "deepseek-ai/DeepSeek-Coder-V2-Instruct" -> it doesn't work (error on StopIteration)
154
+ #repo_id = "meta-llama/CodeLlama-34b-Instruct-hf" -> it doesn't work (error on StopIteration)
155
  chat = ChatHuggingFace(
156
  #llm=HuggingFaceEndpoint(
157
  # endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",