refactor: 尝试去掉nvidia依赖

This commit is contained in:
liqiang-fit2cloud 2024-03-20 12:57:16 +08:00
parent 8fd50f091b
commit 107d85b7b2

View File

@ -19,10 +19,7 @@ diskcache = "^5.6.3"
pillow = "^10.2.0"
filetype = "^1.2.0"
chardet = "^5.2.0"
torch = [
{ url = "https://download.pytorch.org/whl/cpu/torch-2.2.1%2Bcpu-cp311-cp311-linux_x86_64.whl", markers = "sys_platform == 'linux'"},
{ url = "https://download.pytorch.org/whl/cpu/torch-2.2.1-cp311-none-macosx_11_0_arm64.whl", markers = "sys_platform == 'darwin' and platform_machine == 'arm64'"}
]
torch = { url = "https://download.pytorch.org/whl/cpu/torch-2.2.1%2Bcpu-cp311-cp311-linux_x86_64.whl", markers = "sys_platform == 'linux'"}
sentence-transformers = "^2.2.2"
blinker = "^1.6.3"
openai = "^1.13.3"