wishwakankanamg commited on
Commit
54868b2
·
1 Parent(s): ffa0300
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -155,10 +155,10 @@ if not hf_token:
155
  raise ValueError("Hugging Face API token (HF_TOKEN) not found in environment variables.")
156
 
157
  # Build graph function
158
- def build_graph(provider: str = "groq"):
159
  """Build the graph"""
160
 
161
- repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
162
 
163
  if not hf_token:
164
  raise ValueError("HF_TOKEN environment variable not set. It's required for Hugging Face provider.")
 
155
  raise ValueError("Hugging Face API token (HF_TOKEN) not found in environment variables.")
156
 
157
  # Build graph function
158
+ def build_graph(provider: str = "huggingface"):
159
  """Build the graph"""
160
 
161
+ repo_id = "togethercomputer/evo-1-131k-base"
162
 
163
  if not hf_token:
164
  raise ValueError("HF_TOKEN environment variable not set. It's required for Hugging Face provider.")