From dcabda42d1db925c6274da5a2f50410a16f7e3ca Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Thu, 17 Oct 2024 23:04:11 +0200 Subject: [PATCH] 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 --- backend/python/mamba/requirements-cpu.txt | 2 +- backend/python/mamba/requirements-cublas11.txt | 2 +- backend/python/mamba/requirements-cublas12.txt | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/python/mamba/requirements-cpu.txt b/backend/python/mamba/requirements-cpu.txt index 39dab0fd..b4f1261f 100644 --- a/backend/python/mamba/requirements-cpu.txt +++ b/backend/python/mamba/requirements-cpu.txt @@ -1,2 +1,2 @@ -torch +torch==2.4.1 transformers \ No newline at end of file diff --git a/backend/python/mamba/requirements-cublas11.txt b/backend/python/mamba/requirements-cublas11.txt index 7048a14f..ed0d4df5 100644 --- a/backend/python/mamba/requirements-cublas11.txt +++ b/backend/python/mamba/requirements-cublas11.txt @@ -1,3 +1,3 @@ --extra-index-url https://download.pytorch.org/whl/cu118 -torch +torch==2.4.1+cu118 transformers \ No newline at end of file diff --git a/backend/python/mamba/requirements-cublas12.txt b/backend/python/mamba/requirements-cublas12.txt index 39dab0fd..b4f1261f 100644 --- a/backend/python/mamba/requirements-cublas12.txt +++ b/backend/python/mamba/requirements-cublas12.txt @@ -1,2 +1,2 @@ -torch +torch==2.4.1 transformers \ No newline at end of file