[parser] support omegaconf (#7793)

2026-06-22 07:08:57 +08:00 · 2025-04-21 23:30:30 +08:00
parent bd7bc31c79
commit 416853dd25
25 changed files with 62 additions and 94 deletions
--- a/src/llamafactory/train/kto/workflow.py
+++ b/src/llamafactory/train/kto/workflow.py
@@ -59,9 +59,6 @@ def run_kto(
    else:
        ref_model = create_ref_model(model_args, finetuning_args)

-    # Update arguments
-    training_args.remove_unused_columns = False  # important for multimodal and pairwise dataset
-
    # Initialize our Trainer
    trainer = CustomKTOTrainer(
        model=model,