IamRulo commited on
Commit
98521b0
·
verified ·
1 Parent(s): 8c30f51

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -691,7 +691,8 @@ def build_graph(provider: str = "huggingface"):
691
  llm = ChatHuggingFace(
692
  llm=HuggingFaceEndpoint(
693
  #repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
694
- endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
 
695
  #task="text-generation", # for chat‐style use “text-generation”
696
  #max_new_tokens=1024,
697
  #do_sample=False,
 
691
  llm = ChatHuggingFace(
692
  llm=HuggingFaceEndpoint(
693
  #repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
694
+ #endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
695
+ endpoint_url="https://api-inference.huggingface.co/models/Qwen/Qwen3-4B,
696
  #task="text-generation", # for chat‐style use “text-generation”
697
  #max_new_tokens=1024,
698
  #do_sample=False,