LocalAI/backend/python/rerankers/requirements-hipblas.txt

--extra-index-url https://download.pytorch.org/whl/rocm6.0
transformers
accelerate
torch
rerankers[transformers]
fix: install pytorch from proper index for hipblas builds (#2413) Signed-off-by: Chris Jowett <421501+cryptk@users.noreply.github.com> 2024-05-26 18:05:52 +00:00			`--extra-index-url https://download.pytorch.org/whl/rocm6.0`
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`
			`accelerate`
			`torch`
			`rerankers[transformers]`