Initialize qwen-image on CPU

This commit is contained in:
feng0w0
2026-02-05 11:54:36 +08:00
parent 6886f7ba35
commit fccb1ecdd7
4 changed files with 8 additions and 4 deletions

View File

@@ -16,4 +16,5 @@ accelerate launch --config_file examples/qwen_image/model_training/full/accelera
--output_path "./models/train/Qwen-Image-Edit-2509_full" \
--trainable_models "dit" \
--use_gradient_checkpointing \
--find_unused_parameters
--find_unused_parameters \
--initialize_model_on_cpu