文档环境安装上手

2026-03-18 22:08:13 +00:00 · 2025-12-27 16:11:27 +08:00
parent 01234e59c0
commit 9834d72e4d
3 changed files with 2 additions and 4 deletions
--- a/docs/zh/Pipeline_Usage/Setup_NPU_样例1.md
+++ b/docs/zh/Pipeline_Usage/Setup_NPU_样例1.md
--- a/docs/zh/Pipeline_Usage/Setup_NPU_样例2.md
+++ b/docs/zh/Pipeline_Usage/Setup_NPU_样例2.md
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,6 +10,8 @@ authors = [{name = "ModelScope Team"}]
 license = {text = "Apache-2.0"}
 requires-python = ">=3.10"
 dependencies = [
+    "torch>=2.0.0",
+    "torchvision",
    "transformers",
    "imageio",
    "imageio[ffmpeg]",
@@ -31,10 +33,6 @@ classifiers = [
 ]

 [project.optional-dependencies]
-gpu = [
-    "torch>=2.0.0",
-    "torchvision"
-]
 npu_aarch64 = [
    "torch==2.7.1",
    "torch-npu==2.7.1",