LocalAI/backend/python/transformers-musicgen/requirements-cublas11.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

4 lines
99 B
Plaintext

--extra-index-url https://download.pytorch.org/whl/cu118
transformers
accelerate
torch==2.4.1+cu118