Spaces:
Sleeping
Sleeping
EtienneB
commited on
Commit
·
294f2c0
1
Parent(s):
85ecabb
updated model
Browse files
agent.py
CHANGED
|
@@ -47,7 +47,13 @@ def build_graph():
|
|
| 47 |
|
| 48 |
# First create the HuggingFaceEndpoint
|
| 49 |
llm_endpoint = HuggingFaceEndpoint(
|
| 50 |
-
repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
|
| 52 |
temperature=0.1, # Lower temperature for more consistent responses
|
| 53 |
max_new_tokens=1024,
|
|
|
|
| 47 |
|
| 48 |
# First create the HuggingFaceEndpoint
|
| 49 |
llm_endpoint = HuggingFaceEndpoint(
|
| 50 |
+
# repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
|
| 51 |
+
repo_id="mistralai/Mistral-7B-Instruct-v0.2",
|
| 52 |
+
# Other models to try:
|
| 53 |
+
# "meta-llama/Llama-2-7b-chat-hf"
|
| 54 |
+
# "google/gemma-7b-it"
|
| 55 |
+
# "mosaicml/mpt-7b-instruct"
|
| 56 |
+
# "tiiuae/falcon-7b-instruct"
|
| 57 |
huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
|
| 58 |
temperature=0.1, # Lower temperature for more consistent responses
|
| 59 |
max_new_tokens=1024,
|