LocalAI/backend/python/rerankers/requirements-hipblas.txt
Ettore Di Giacinto 3e77a17b26
fix(dependencies): pin pytorch version (#3872)
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
2024-10-18 09:11:59 +02:00

5 lines
127 B
Plaintext

--extra-index-url https://download.pytorch.org/whl/rocm6.0
transformers
accelerate
torch==2.4.1+rocm6.0
rerankers[transformers]