Merge pull request #1128 from modelscope/amd_install

update installation instructions for AMD

README.md

@@ -189,21 +189,7 @@ cd DiffSynth-Studio
 pip install -e .
 ```
 
-<details>
+For more installation methods and instructions for non-NVIDIA GPUs, please refer to the [Installation Guide](/docs/en/Pipeline_Usage/Setup.md).
-<summary>Other installation methods</summary>
-
-Install from PyPI (version updates may be delayed; for latest features, install from source)
-
-```
-pip install diffsynth
-```
-
-If you meet problems during installation, they might be caused by upstream dependencies. Please check the docs of these packages:
-
-* [torch](https://pytorch.org/get-started/locally/)
-* [sentencepiece](https://github.com/google/sentencepiece)
-* [cmake](https://cmake.org)
-* [cupy](https://docs.cupy.dev/en/stable/install.html)
 
 </details>
 

README_zh.md

@@ -189,21 +189,7 @@ cd DiffSynth-Studio
 pip install -e .
 ```
 
-<details>
+更多安装方式，以及非 NVIDIA GPU 的安装，请参考[安装文档](/docs/zh/Pipeline_Usage/Setup.md)。
-<summary>其他安装方式</summary>
-
-从 pypi 安装（存在版本更新延迟，如需使用最新功能，请从源码安装）
-
-```
-pip install diffsynth
-```
-
-如果在安装过程中遇到问题，可能是由上游依赖包导致的，请参考这些包的文档：
-
-* [torch](https://pytorch.org/get-started/locally/)
-* [sentencepiece](https://github.com/google/sentencepiece)
-* [cmake](https://cmake.org)
-* [cupy](https://docs.cupy.dev/en/stable/install.html)
 
 </details>
 

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)

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)