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

This reverts commit ef7a783ac7.
This commit is contained in:
liqiang-fit2cloud 2024-03-20 19:30:57 +08:00
parent ef7a783ac7
commit 6bf3ccfa71
2 changed files with 3 additions and 8 deletions

View File

@ -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

View File

@ -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"