SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 07:40:12 +01:00
Code Issues Projects Releases Packages Activity

feat: move torch and transformers to local group (#1172)

Browse source
This commit is contained in:
Pablo Orgaz 2023-11-06 14:24:16 +01:00 • committed by GitHub
parent ad512e3c42
commit 0d677e10b9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 566 additions and 560 deletions
Download patch file Download diff file Expand all files Collapse all files

10
pyproject.toml
View file

@ -15,9 +15,6 @@ pypdf = "^3.16.2"
llama-index = "0.8.47"
chromadb = "^0.4.13"
watchdog = "^3.0.0"
transformers = "^4.34.0"
# https://stackoverflow.com/questions/76327419/valueerror-libcublas-so-0-9-not-found-in-the-system-path
torch = ">=2.0.0, !=2.0.1, !=2.1.0"
[tool.poetry.group.dev.dependencies]
black = "^22"
@ -38,9 +35,12 @@ gradio = "^3.45.2"
[tool.poetry.group.local]
optional = true
[tool.poetry.group.local.dependencies]
sentence-transformers = "^2.2.2"
numpy = "1.26.0"
llama-cpp-python = "^0.2.11"
numpy = "1.26.0"
sentence-transformers = "^2.2.2"
# https://stackoverflow.com/questions/76327419/valueerror-libcublas-so-0-9-not-found-in-the-system-path
torch = ">=2.0.0, !=2.0.1, !=2.1.0"
transformers = "^4.34.0"
[build-system]
requires = ["poetry-core>=1.0.0"]