Merge pull request #1167 from Feng0w0/install_env

Docs:Supplement NPU environment installation document
2026-03-18 22:08:13 +00:00 · 2026-01-12 10:07:30 +08:00
parent de0aa946f7 9cc1697d4d
commit 8cc3bece6d
3 changed files with 30 additions and 8 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,5 +36,17 @@ classifiers = [
 where = ["./"]
 include = ["diffsynth"]

+[project.optional-dependencies]
+npu_aarch64 = [
+    "torch==2.7.1",
+    "torch-npu==2.7.1",
+    "torchvision==0.22.1"
+]
+npu = [
+    "torch==2.7.1+cpu",
+    "torch-npu==2.7.1",
+    "torchvision==0.22.1+cpu"
+]
+
 [tool.setuptools]
 include-package-data = true