diff --git a/docs/en/Pipeline_Usage/GPU_support.md b/docs/en/Pipeline_Usage/GPU_support.md index bfaa7cefa..e27d23ad0 100644 --- a/docs/en/Pipeline_Usage/GPU_support.md +++ b/docs/en/Pipeline_Usage/GPU_support.md @@ -90,4 +90,5 @@ Set 0 or not set: indicates not enabling the binding function | Model | Parameter | Note | |----------------|---------------------------|-------------------| | Wan 14B series | --initialize_model_on_cpu | The 14B model needs to be initialized on the CPU | -| Qwen-Image series | --initialize_model_on_cpu | The model needs to be initialized on the CPU | \ No newline at end of file +| Qwen-Image series | --initialize_model_on_cpu | The model needs to be initialized on the CPU | +| Z-Image series | --enable_npu_patch | Using NPU fusion operator to replace the corresponding operator in Z-image model to improve the performance of the model on NPU | \ No newline at end of file diff --git a/docs/en/Pipeline_Usage/Setup.md b/docs/en/Pipeline_Usage/Setup.md index 153f84ac6..2e45668ce 100644 --- a/docs/en/Pipeline_Usage/Setup.md +++ b/docs/en/Pipeline_Usage/Setup.md @@ -37,9 +37,9 @@ pip install torch torchvision --index-url https://download.pytorch.org/whl/rocm6 git clone https://github.com/modelscope/DiffSynth-Studio.git cd DiffSynth-Studio # aarch64/ARM - pip install -e .[npu_aarch64] --extra-index-url "https://download.pytorch.org/whl/cpu" + pip install -e .[npu_aarch64] # x86 - pip install -e .[npu] + pip install -e .[npu] --extra-index-url "https://download.pytorch.org/whl/cpu" When using Ascend NPU, please replace `"cuda"` with `"npu"` in your Python code. For details, see [NPU Support](../Pipeline_Usage/GPU_support.md#ascend-npu). diff --git a/docs/zh/Pipeline_Usage/GPU_support.md b/docs/zh/Pipeline_Usage/GPU_support.md index c56704e07..9760aea97 100644 --- a/docs/zh/Pipeline_Usage/GPU_support.md +++ b/docs/zh/Pipeline_Usage/GPU_support.md @@ -89,4 +89,5 @@ export CPU_AFFINITY_CONF=1 | 模型 | 参数 | 备注 | |-----------|------|-------------------| | Wan 14B系列 | --initialize_model_on_cpu | 14B模型需要在cpu上进行初始化 | -| Qwen-Image系列 | --initialize_model_on_cpu | 模型需要在cpu上进行初始化 | \ No newline at end of file +| Qwen-Image系列 | --initialize_model_on_cpu | 模型需要在cpu上进行初始化 | +| Z-Image 系列 | --enable_npu_patch | 使用NPU融合算子来替换Z-image模型中的对应算子以提升模型在NPU上的性能 | \ No newline at end of file diff --git a/docs/zh/Pipeline_Usage/Setup.md b/docs/zh/Pipeline_Usage/Setup.md index 8b2b323ee..13a2caeb8 100644 --- a/docs/zh/Pipeline_Usage/Setup.md +++ b/docs/zh/Pipeline_Usage/Setup.md @@ -37,9 +37,9 @@ pip install torch torchvision --index-url https://download.pytorch.org/whl/rocm6 git clone https://github.com/modelscope/DiffSynth-Studio.git cd DiffSynth-Studio # aarch64/ARM - pip install -e .[npu_aarch64] --extra-index-url "https://download.pytorch.org/whl/cpu" + pip install -e .[npu_aarch64] # x86 - pip install -e .[npu] + pip install -e .[npu] --extra-index-url "https://download.pytorch.org/whl/cpu" 使用 Ascend NPU 时,请将 Python 代码中的 `"cuda"` 改为 `"npu"`,详见[NPU 支持](../Pipeline_Usage/GPU_support.md#ascend-npu)。