diff --git a/docs/zh/Pipeline_Usage/Setup_NPU_doc.md b/docs/zh/Pipeline_Usage/Setup_NPU_样例1.md similarity index 100% rename from docs/zh/Pipeline_Usage/Setup_NPU_doc.md rename to docs/zh/Pipeline_Usage/Setup_NPU_样例1.md diff --git a/docs/zh/Pipeline_Usage/Setup_NPU_pip.md b/docs/zh/Pipeline_Usage/Setup_NPU_样例2.md similarity index 100% rename from docs/zh/Pipeline_Usage/Setup_NPU_pip.md rename to docs/zh/Pipeline_Usage/Setup_NPU_样例2.md diff --git a/pyproject.toml b/pyproject.toml index 3b165d3..7597d81 100644 --- 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",