Update agent.py
Browse files
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",
|