Chris4K commited on
Commit
1eb5c78
1 Parent(s): 953ea93
Files changed (1) hide show
  1. model/custom_agent.py +3 -3
model/custom_agent.py CHANGED
@@ -137,8 +137,8 @@ class CustomHfAgent(Agent):
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,5 +148,5 @@ class CustomHfAgent(Agent):
148
  "content": prompt,
149
  }
150
  ]
151
- prompt = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)
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