kyy/llm_trainer
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add kto

Browse Source
This commit is contained in:
enji.zhou committed 2024-05-17 13:09:17 +08:00
1 parent 84415492bf
commit db1d5a4f51
14 files changed
+5923 -8
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/llamafactory/extras/constants.py
+1
View File
@@ -45,6 +45,7 @@ TRAINING_STAGES = {
"Reward Modeling": "rm",
"PPO": "ppo",
"DPO": "dpo",
"KTO": "kto",
"ORPO": "orpo",
"Pre-Training": "pt",
}