support edit_image_auto_resize

2026-03-23 17:38:10 +00:00 · 2025-08-20 12:36:26 +08:00
parent 9d0683df25
commit 29b12b2f4e
5 changed files with 34 additions and 11 deletions
--- a/examples/qwen_image/model_training/train.py
+++ b/examples/qwen_image/model_training/train.py
@@ -78,6 +78,7 @@ class QwenImageTrainingModule(DiffusionTrainingModule):
            "rand_device": self.pipe.device,
            "use_gradient_checkpointing": self.use_gradient_checkpointing,
            "use_gradient_checkpointing_offload": self.use_gradient_checkpointing_offload,
+            "edit_image_auto_resize": True,
        }
        
        # Extra inputs