comment
Browse files- model/custom_agent.py +3 -3
model/custom_agent.py
CHANGED
@@ -137,8 +137,8 @@ class CustomHfAgent(Agent):
|
|
137 |
"content": prompt,
|
138 |
}
|
139 |
]
|
140 |
-
|
141 |
-
|
142 |
else:
|
143 |
prompt = self.run_prompt_template.replace("<<all_tools>>", description)
|
144 |
prompt = prompt.replace("<<prompt>>", task)
|
@@ -148,5 +148,5 @@ class CustomHfAgent(Agent):
|
|
148 |
"content": prompt,
|
149 |
}
|
150 |
]
|
151 |
-
|
152 |
return prompt
|
|
|
137 |
"content": prompt,
|
138 |
}
|
139 |
]
|
140 |
+
# prompt = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)
|
141 |
+
## prompt
|
142 |
else:
|
143 |
prompt = self.run_prompt_template.replace("<<all_tools>>", description)
|
144 |
prompt = prompt.replace("<<prompt>>", task)
|
|
|
148 |
"content": prompt,
|
149 |
}
|
150 |
]
|
151 |
+
# prompt = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)
|
152 |
return prompt
|