15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

refactor: 尝试去掉nvidia依赖

This commit is contained in:
liqiang-fit2cloud 2024-03-21 10:17:44 +08:00
parent 6bf3ccfa71
commit a030275326
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/build-and-push.yml vendored
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

7
pyproject.toml
View File

@ -18,7 +18,12 @@ diskcache = "^5.6.3"
pillow = "^10.2.0"
filetype = "^1.2.0"
chardet = "^5.2.0"
torch = {version = "^2.2.1+cpu", source = "pytorch"}
torch = [
{ version = "^2.2.1+cpu", markers = "platform_system == 'Linux' and platform_machine == 'x86_64'", source = "pytorch" },
{ version = "^2.2.1", markers = "platform_system == 'Linux' and platform_machine == 'aarch64'", source = "pytorch" },
{ version = "^2.2.1+cpu", markers = "platform_system == 'Windows'", source = "pytorch" },
{ version = "^2.2.1", markers = "platform_system == 'Darwin'", source = "pytorch" }
]
sentence-transformers = "^2.2.2"
blinker = "^1.6.3"
openai = "^1.13.3"