fix: set add_generation_prompt to False when apply chat template (#2859) [skip ci]

This commit is contained in:
NanoCode012
2025-07-08 22:00:44 +07:00
committed by GitHub
parent b237c8a3f3
commit 78bff4925e

View File

@@ -50,7 +50,7 @@ class MultiModalChatDataCollator(DataCollatorMixin):
# This method requires transformers>=4.49.0
result = self.processing_strategy.processor.apply_chat_template(
example["messages"],
add_generation_prompt=True,
add_generation_prompt=False,
tokenize=True,
return_tensors="pt",
padding=True,