merge data process to training script

2026-03-18 22:08:13 +00:00 · 2025-09-04 15:18:56 +08:00
parent cb8de6be1b
commit 144365b07d
6 changed files with 35 additions and 147 deletions
--- a/examples/flux/model_training/train.py
+++ b/examples/flux/model_training/train.py
@@ -25,7 +25,7 @@ class FluxTrainingModule(DiffusionTrainingModule):
        
        # Training mode
        self.switch_pipe_to_training_mode(
-            self, self.pipe, trainable_models,
+            self.pipe, trainable_models,
            lora_base_model, lora_target_modules, lora_rank, lora_checkpoint=lora_checkpoint,
            enable_fp8_training=False,
        )