Shago commited on
Commit
a8a9435
·
verified ·
1 Parent(s): 72f1955

Update agents/agents_nodes.py

Browse files
Files changed (1) hide show
  1. agents/agents_nodes.py +2 -1
agents/agents_nodes.py CHANGED
@@ -24,9 +24,10 @@ from tools.financial_tools import time_value_tool
24
  import os
25
  from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
26
 
 
27
 
28
  llm_endpoint = HuggingFaceEndpoint(
29
- endpoint_url="https://api-inference.huggingface.co/models/google/gemma-2b-it",
30
  huggingfacehub_api_token=os.environ["HF_TOKEN"], # Set in Space secrets
31
  max_new_tokens=500,
32
  temperature=0
 
24
  import os
25
  from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
26
 
27
+ repo_id="google/gemma-2b-it"
28
 
29
  llm_endpoint = HuggingFaceEndpoint(
30
+ repo_id=repo_id
31
  huggingfacehub_api_token=os.environ["HF_TOKEN"], # Set in Space secrets
32
  max_new_tokens=500,
33
  temperature=0