fix: fix pytorch version to avoid wheel bug (#1123)

* fix: fix pytorch version

* fix: settings env var regex and split

* fix: add models folder for docker user
This commit is contained in:
Pablo Orgaz 2023-10-27 20:27:40 +02:00 committed by GitHub
parent 895588b82a
commit 24cfddd60f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 1150 additions and 480 deletions

View file

@ -17,7 +17,8 @@ llama-index = "0.8.47"
chromadb = "^0.4.13"
watchdog = "^3.0.0"
transformers = "^4.34.0"
torch = "^2.1.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"