Shago commited on
Commit
ea26cee
·
verified ·
1 Parent(s): 67e33c2

Update agents/agents_nodes.py

Browse files
Files changed (1) hide show
  1. agents/agents_nodes.py +1 -1
agents/agents_nodes.py CHANGED
@@ -29,7 +29,7 @@ from langchain_core.messages import AIMessage
29
  # Initialize Hugging Face endpoint (replace with your model)
30
  HF_MODEL = "google/gemma-2b-it"
31
  llm_endpoint = HuggingFaceEndpoint(
32
- endpoint_url=f"https://api-inference.huggingface.co/models/{HF_MODEL}",
33
  huggingfacehub_api_token=os.environ["HF_TOKEN"],
34
  max_new_tokens=500,
35
  temperature=0
 
29
  # Initialize Hugging Face endpoint (replace with your model)
30
  HF_MODEL = "google/gemma-2b-it"
31
  llm_endpoint = HuggingFaceEndpoint(
32
+ endpoint_url=f"https://huggingface.co/models/{HF_MODEL}",
33
  huggingfacehub_api_token=os.environ["HF_TOKEN"],
34
  max_new_tokens=500,
35
  temperature=0