Revert "refactor: 尝试去掉nvidia依赖"

This reverts commit ef7a783.
End-Q · Mar 20, 2024 · 6bf3ccf · 6bf3ccf
1 parent ef7a783
commit 6bf3ccf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/.github/workflows/build-and-push.yml b/.github/workflows/build-and-push.yml
@@ -14,8 +14,8 @@ on:
         type: choice
         options:
           - linux/amd64
-          - linux/arm64
-          - linux/amd64,linux/arm64
+          #- linux/arm64
+          #- linux/amd64,linux/arm64
       registry:
         description: 'Push To Registry'
         required: true

diff --git a/pyproject.toml b/pyproject.toml
@@ -18,12 +18,7 @@ diskcache = "^5.6.3"
 pillow = "^10.2.0"
 filetype = "^1.2.0"
 chardet = "^5.2.0"
-torch = [
-    { version = "^2.2.1+cpu", markers = "sys_platform == 'linux' and (platform_machine == 'x86_64' or platform_machine == 'amd64')", source = "pytorch" },
-    { version = "^2.2.1+cpu", markers = "sys_platform == 'win32'", source = "pytorch" },
-    { version = "^2.2.1", markers = "sys_platform == 'linux' and (platform_machine == 'arm64' or platform_machine == 'aarch64')", source = "pytorch" },
-    { version = "^2.2.1", markers = "sys_platform == 'darwin'", source = "pytorch" }
-]
+torch = {version = "^2.2.1+cpu", source = "pytorch"}
 sentence-transformers = "^2.2.2"
 blinker = "^1.6.3"
 openai = "^1.13.3"