From 6bf3ccfa71b692339e98cc7440cf03a98327f948 Mon Sep 17 00:00:00 2001
From: liqiang-fit2cloud <liqiang@fit2cloud.com>
Date: Wed, 20 Mar 2024 19:30:57 +0800
Subject: [PATCH] =?UTF-8?q?Revert=20"refactor:=20=E5=B0=9D=E8=AF=95?=
 =?UTF-8?q?=E5=8E=BB=E6=8E=89nvidia=E4=BE=9D=E8=B5=96"?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

This reverts commit ef7a783ac786f7eb43dadbbe210be909f835583c.
---
 .github/workflows/build-and-push.yml | 4 ++--
 pyproject.toml                       | 7 +------
 2 files changed, 3 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/build-and-push.yml b/.github/workflows/build-and-push.yml
index 7a6f51de8..7c089d884 100644
--- 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
index b39964058..268c23c96 100644
--- 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"