support qwen-image blockwise controlnet training

This commit is contained in:
Artiprocher
2025-08-15 18:41:01 +08:00
parent 024fdad76d
commit 01a1f48f70
17 changed files with 269 additions and 15 deletions

View File

@@ -9,4 +9,5 @@ accelerate launch --config_file examples/qwen_image/model_training/full/accelera
--remove_prefix_in_ckpt "pipe.dit." \
--output_path "./models/train/Qwen-Image_full" \
--trainable_models "dit" \
--use_gradient_checkpointing
--use_gradient_checkpointing \
--find_unused_parameters