Spaces:
Sleeping
Sleeping
Model tunning
Browse filesDecreased max tokens and steps.
app.py
CHANGED
|
@@ -82,7 +82,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
|
|
| 82 |
final_answer = FinalAnswerTool()
|
| 83 |
search_tool = DuckDuckGoSearchTool()
|
| 84 |
model = HfApiModel(
|
| 85 |
-
max_tokens=
|
| 86 |
temperature=0.5,
|
| 87 |
model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',
|
| 88 |
custom_role_conversions=None,
|
|
@@ -98,7 +98,7 @@ with open("prompts.yaml", 'r') as stream:
|
|
| 98 |
agent = CodeAgent(
|
| 99 |
model=model,
|
| 100 |
tools=[execute_shell_command, image_generation_tool, search_tool, final_answer],
|
| 101 |
-
max_steps=
|
| 102 |
verbosity_level=1,
|
| 103 |
grammar=None,
|
| 104 |
planning_interval=None,
|
|
|
|
| 82 |
final_answer = FinalAnswerTool()
|
| 83 |
search_tool = DuckDuckGoSearchTool()
|
| 84 |
model = HfApiModel(
|
| 85 |
+
max_tokens=1024,
|
| 86 |
temperature=0.5,
|
| 87 |
model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',
|
| 88 |
custom_role_conversions=None,
|
|
|
|
| 98 |
agent = CodeAgent(
|
| 99 |
model=model,
|
| 100 |
tools=[execute_shell_command, image_generation_tool, search_tool, final_answer],
|
| 101 |
+
max_steps=3,
|
| 102 |
verbosity_level=1,
|
| 103 |
grammar=None,
|
| 104 |
planning_interval=None,
|