Update agent.py
Browse files
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:
|