From 3140199c961520bd773db41bd4ea79157f37c7a4 Mon Sep 17 00:00:00 2001 From: feng0w0 Date: Tue, 27 Jan 2026 15:33:42 +0800 Subject: [PATCH] [feature]:Add adaptation of all models to zero3 --- .../special/npu_training/Qwen-Image-Edit-2509-NPU.sh | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/qwen_image/model_training/special/npu_training/Qwen-Image-Edit-2509-NPU.sh b/examples/qwen_image/model_training/special/npu_training/Qwen-Image-Edit-2509-NPU.sh index 24a7a58..4892228 100644 --- a/examples/qwen_image/model_training/special/npu_training/Qwen-Image-Edit-2509-NPU.sh +++ b/examples/qwen_image/model_training/special/npu_training/Qwen-Image-Edit-2509-NPU.sh @@ -1,4 +1,7 @@ # This script was tested using zero3 and on 8*910B(NPU) +export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True +export CPU_AFFINITY_CONF=1 + accelerate launch --config_file examples/qwen_image/model_training/full/accelerate_config_zero3.yaml examples/qwen_image/model_training/train.py \ --dataset_base_path data/example_image_dataset \ --dataset_metadata_path data/example_image_dataset/metadata_qwen_imgae_edit_multi.json \