EtienneB commited on
Commit
ba0e6f8
·
1 Parent(s): d49a17e

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -144,8 +144,9 @@ def build_graph():
144
  # First create the HuggingFaceEndpoint
145
  llm_endpoint = HuggingFaceEndpoint(
146
  # repo_id="Qwen/Qwen2.5-14B-Instruct",
147
- repo_id="HuggingFaceH4/mistral-7b-anthropic",
148
  # repo_id="Qwen/Qwen2.5-Coder-32B-Instruct", # this one is poor
 
149
  huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
150
  temperature=0.1, # Maximum determinism
151
  max_new_tokens=512, # Even more restrictive with 128
 
144
  # First create the HuggingFaceEndpoint
145
  llm_endpoint = HuggingFaceEndpoint(
146
  # repo_id="Qwen/Qwen2.5-14B-Instruct",
147
+ # repo_id="HuggingFaceH4/mistral-7b-anthropic", # wrong id?
148
  # repo_id="Qwen/Qwen2.5-Coder-32B-Instruct", # this one is poor
149
+ repo_id="meta-llama/Llama-3.1-8B",
150
  huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
151
  temperature=0.1, # Maximum determinism
152
  max_new_tokens=512, # Even more restrictive with 128