update doc

2026-03-23 09:28:12 +00:00 · 2025-11-10 10:05:19 +08:00
parent ceb473efc0
commit eacec13309
13 changed files with 489 additions and 8 deletions
--- a/examples/qwen_image/model_training/train.py
+++ b/examples/qwen_image/model_training/train.py
@@ -72,6 +72,7 @@ class QwenImageTrainingModule(DiffusionTrainingModule):
    
    def forward(self, data, inputs=None):
        if self.fp8_models is not None:
+            # TODO: remove it
            self.pipe.flush_vram_management_device(self.pipe.device)
        if inputs is None: inputs = self.get_pipeline_inputs(data)
        inputs = self.transfer_data_to_device(inputs, self.pipe.device, self.pipe.torch_dtype)