IamRulo commited on
Commit
58e696d
·
verified ·
1 Parent(s): ea09f47

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +4 -4
agent.py CHANGED
@@ -692,12 +692,12 @@ def build_graph(provider: str = "huggingface"):
692
  llm=HuggingFaceEndpoint(
693
  repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
694
  task="text-generation", # for chat‐style use “text-generation”
695
- max_new_tokens=1024,
696
- do_sample=False,
697
- repetition_penalty=1.03,
698
  temperature=0,
699
  ),
700
- verbose=True,
701
  )
702
 
703
  else:
 
692
  llm=HuggingFaceEndpoint(
693
  repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
694
  task="text-generation", # for chat‐style use “text-generation”
695
+ #max_new_tokens=1024,
696
+ #do_sample=False,
697
+ #repetition_penalty=1.03,
698
  temperature=0,
699
  ),
700
+ #verbose=True,
701
  )
702
 
703
  else: