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,
|