[assets] update readme (#7644)

2025-04-09 01:06:06 +08:00
parent 349c56c51c
commit 1abd71b551
8 changed files with 17 additions and 7 deletions
--- a/src/llamafactory/train/kto/trainer.py
+++ b/src/llamafactory/train/kto/trainer.py
@@ -147,6 +147,9 @@ class CustomKTOTrainer(KTOTrainer):
        if "pixel_values" in batch:
            model_inputs["pixel_values"] = batch["pixel_values"]

+        if "image_sizes" in batch:
+            model_inputs["image_sizes"] = batch["image_sizes"]
+
        if "image_grid_thw" in batch:
            model_inputs["image_grid_thw"] = batch["image_grid_thw"]