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

remove unused field for chat_template.default for DPO training (#2755) [skip ci]

Browse Source 
* remove unused field for chat_template.default

"messages" field present in final dataset causes issues with DPO
training otherwise

* lint and fix tests for new return value

* remove unused field for chat_template.default

"messages" field present in final dataset causes issues with DPO
training otherwise

lint and fix tests for new return value

fix for updated expected fields for dpo

remove unused field for chat_template.default

"messages" field present in final dataset causes issues with DPO
training otherwise

fix test still expecting "messages" field

* chore: lint

---------

Co-authored-by: Wing Lian <wing@axolotl.ai>
This commit is contained in:
Timofey Klyubin
2025-06-05 10:22:58 -04:00
committed by GitHub
parent e8e45b3441
commit 4440b4a1ce
3 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/prompt_strategies/dpo/chat_template.py
View File

@@ -91,4 +91,4 @@ def default(
return result
return transform_fn
return transform_fn, {"remove_columns": [field_messages]}