build: Pin down triton (#11938)

Signed-off-by: oliver könig <okoenig@nvidia.com>
NVIDIA · Jan 23, 2025 · eb892ae · eb892ae
1 parent 116b5cd
commit eb892ae
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/reinstall.sh b/reinstall.sh
@@ -65,6 +65,7 @@ if [[ "$HEAVY_DEPS" == "TRUE" ]]; then
     "unstructured==0.14.9"
     "git+https://github.com/Dao-AILab/causal-conv1d.git@${CAUSAL_CONV_TAG}"
     "git+https://github.com/state-spaces/mamba.git@${MAMBA_TAG}"
+    "triton==3.1.0"
   )
 
   pip install --no-cache-dir -r tools/ctc_segmentation/requirements.txt