EtienneB commited on
Commit
294f2c0
·
1 Parent(s): 85ecabb

updated model

Browse files
Files changed (1) hide show
  1. agent.py +7 -1
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,