tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
NanoCode012
2025-07-08 22:00:44 +07:00
committed by GitHub
parent b237c8a3f3
commit 78bff4925e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/collators/mm_chat.py
View File

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