[model][NPU] Add NPU fusion operator patch to Zimage model to improve performance

This commit is contained in:
feng0w0
2026-02-03 19:50:21 +08:00
parent ffb7a138f7
commit 051b957adb
5 changed files with 66 additions and 13 deletions

View File

@@ -13,4 +13,5 @@ accelerate launch --config_file examples/z_image/model_training/full/accelerate_
--output_path "./models/train/Z-Image-Turbo_full" \
--trainable_models "dit" \
--use_gradient_checkpointing \
--dataset_num_workers 8
--dataset_num_workers 8 \
--enable_npu_patch