Spaces:
Sleeping
Sleeping
Commit
·
4ad0811
1
Parent(s):
30104af
HuggingFace RepoId
Browse files
agent.py
CHANGED
|
@@ -162,11 +162,9 @@ def build_graph(provider: str = "huggingface"):
|
|
| 162 |
# Groq https://console.groq.com/docs/models
|
| 163 |
llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
| 164 |
elif provider == "huggingface":
|
| 165 |
-
# TODO: Add huggingface endpoint
|
| 166 |
llm = ChatHuggingFace(
|
| 167 |
llm=HuggingFaceEndpoint(
|
| 168 |
-
|
| 169 |
-
temperature=0,
|
| 170 |
),
|
| 171 |
)
|
| 172 |
else:
|
|
|
|
| 162 |
# Groq https://console.groq.com/docs/models
|
| 163 |
llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
| 164 |
elif provider == "huggingface":
|
|
|
|
| 165 |
llm = ChatHuggingFace(
|
| 166 |
llm=HuggingFaceEndpoint(
|
| 167 |
+
repo_id = "meta-llama/Llama-2-7b-chat-hf"
|
|
|
|
| 168 |
),
|
| 169 |
)
|
| 170 |
else:
|