Spaces:
Runtime error
Runtime error
Matt commited on
Commit ·
565c08f
1
Parent(s): 4041585
Bugfix
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ def apply_chat_template(template, test_conversation1, test_conversation2):
|
|
| 29 |
for i, conversation_str in enumerate((test_conversation1, test_conversation2)):
|
| 30 |
conversation = json.loads(conversation_str)
|
| 31 |
outputs.append(tokenizer.apply_chat_template(conversation, tokenize=False))
|
| 32 |
-
outputs.append(
|
| 33 |
return tuple(outputs)
|
| 34 |
|
| 35 |
output_names = ["Conversation 1 without generation prompt", "Conversation 1 with generation prompt", "Conversation 2 without generation prompt", "Conversation 2 with generation prompt"]
|
|
|
|
| 29 |
for i, conversation_str in enumerate((test_conversation1, test_conversation2)):
|
| 30 |
conversation = json.loads(conversation_str)
|
| 31 |
outputs.append(tokenizer.apply_chat_template(conversation, tokenize=False))
|
| 32 |
+
outputs.append(tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True))
|
| 33 |
return tuple(outputs)
|
| 34 |
|
| 35 |
output_names = ["Conversation 1 without generation prompt", "Conversation 1 with generation prompt", "Conversation 2 without generation prompt", "Conversation 2 with generation prompt"]
|