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

Merge branch 'main' of github.com:1Panel-dev/MaxKB

This commit is contained in:
shaohuzhang1 2024-03-21 14:50:07 +08:00
parent 28d44ac567 d1e4ba1c3f
commit df896501b6
2 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
installer/Dockerfile
View File

@ -24,6 +24,7 @@ RUN mv /opt/maxkb/app/model/* /opt/maxkb/model && \
apt-get update && apt-get install -y curl && \
pip3 install --upgrade pip && \
pip3 install poetry && \
if [ `uname -m` == "aarch64" ]; then sed -i '/^torch/d' pyproject.toml; fi && \
poetry export -f requirements.txt --output requirements.txt --without-hashes && \
pip3 install --no-cache-dir -r requirements.txt && \
pip3 cache purge && \

7
pyproject.toml
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 = "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" }
]
torch = {version = "^2.2.1+cpu", source = "pytorch"}
sentence-transformers = "^2.2.2"
blinker = "^1.6.3"
openai = "^1.13.3"