LocalAI/backend/python/mamba/requirements-cublas11.txt
Ettore Di Giacinto dcabda42d1
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 23:04:11 +02:00

3 lines
88 B
Plaintext

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