build: Fix triton

Signed-off-by: oliver könig <[email protected]>
NVIDIA · Jan 23, 2025 · 49d9e2a · 49d9e2a
1 parent 9aa2a3d
commit 49d9e2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.ci b/Dockerfile.ci
@@ -39,7 +39,7 @@ ARG MAMBA_TAG=v2.2.2
 
 RUN <<"EOF" bash -ex
 # Mamba dependancy installation
-pip install git+https://github.com/state-spaces/mamba.git@${MAMBA_TAG}
+pip install git+https://github.com/state-spaces/mamba.git@${MAMBA_TAG} "triton=3.1.0"
 git clone --depth 1 --branch ${CAUSAL_CONV_TAG} https://github.com/Dao-AILab/causal-conv1d && \
   cd causal-conv1d && \
   python setup.py install && \