ExternalVendorCode/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2024-12-27 16:08:51 +00:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity
dcabda42d1
LocalAI/backend/python/mamba/requirements-cublas12.txt

2 lines
25 B
Plaintext
Raw Normal View History

fix(mamba): pin torch version (#3871) causal-conv1d supports only torch 2.4.x, not torch 2.5.x Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
2024-10-17 21:04:11 +00:00
torch==2.4.1
fix(python): move accelerate and GPU-specific libs to build-type (#3194) Some of the dependencies in `requirements.txt`, even if generic, pulls down the line CUDA libraries. This changes moves mostly all GPU-specific libs to the build-type, and tries a safer approach. In `requirements.txt` now are listed only "first-level" dependencies, for instance, grpc, but libs-dependencies are moved down to the respective build-type `requirements.txt` to avoid any mixin. This should fix #2737 and #1592. Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
2024-08-07 15:02:32 +00:00
transformers
Reference in New Issue Copy Permalink