Merge pull request #1128 from modelscope/amd_install

update installation instructions for AMD

docs/en/Pipeline_Usage/Setup.md

@@ -14,8 +14,33 @@ Install from PyPI (there may be delays in version updates; for latest features,
 pip install diffsynth
 ```
 
-If you encounter issues during installation, they may be caused by upstream dependency packages. Please refer to the documentation for these packages:
+## GPU/NPU Support
+
+* **NVIDIA GPU**
+
+Install as described above.
+
+* **AMD GPU**
+
+You need to install the `torch` package with ROCm support. Taking ROCm 6.4 (as of the article update date: December 15, 2025) on Linux as an example, run the following command:
+
+```shell
+pip install torch torchvision --index-url https://download.pytorch.org/whl/rocm6.4
+```
+
+* **Ascend NPU**
+
+Ascend NPU support is provided via the `torch-npu` package. Taking version `2.1.0.post17` (as of the article update date: December 15, 2025) as an example, run the following command:
+
+```shell
+pip install torch-npu==2.1.0.post17
+```
+
+## Other Installation Issues
+
+If you encounter issues during installation, they may be caused by upstream dependencies. Please refer to the documentation for these packages:
 
 * [torch](https://pytorch.org/get-started/locally/)
+* [Ascend/pytorch](https://github.com/Ascend/pytorch)
 * [sentencepiece](https://github.com/google/sentencepiece)
-* [cmake](https://cmake.org)
+* [cmake](https://cmake.org)

docs/zh/Pipeline_Usage/Setup.md

@@ -14,8 +14,33 @@ pip install -e .
 pip install diffsynth
 ```
 
+## GPU/NPU 支持
+
+* NVIDIA GPU
+
+按照以上方式安装即可。
+
+* AMD GPU
+
+需安装支持 ROCm 的 `torch` 包，以 ROCm 6.4（本文更新于 2025 年 12 月 15 日）、Linux 系统为例，请运行以下命令
+
+```shell
+pip install torch torchvision --index-url https://download.pytorch.org/whl/rocm6.4
+```
+
+* Ascend NPU
+
+Ascend NPU 通过 `torch-npu` 包提供支持，以 `2.1.0.post17` 版本（本文更新于 2025 年 12 月 15 日）为例，请运行以下命令
+
+```shell
+pip install torch-npu==2.1.0.post17
+```
+
+## 其他安装问题
+
 如果在安装过程中遇到问题，可能是由上游依赖包导致的，请参考这些包的文档：
 
 * [torch](https://pytorch.org/get-started/locally/)
+* [Ascend/pytorch](https://github.com/Ascend/pytorch)
 * [sentencepiece](https://github.com/google/sentencepiece)
 * [cmake](https://cmake.org)