Add triton 2.3.1 to requirements

zackees · Aug 5, 2024 · ea96bf9 · ea96bf9
1 parent 05caaf2
commit ea96bf9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/transcribe_anything/insanley_fast_whisper_reqs.py b/transcribe_anything/insanley_fast_whisper_reqs.py
@@ -40,6 +40,7 @@ def get_environment() -> dict[str, Any]:
         "torchmetrics~=1.3.0",
         "srtranslator==0.2.6",
         "numpy==1.26.4",
+        "triton==2.3.1"
     ]
     if has_nvidia_smi():
         deps.append(f"torch=={TENSOR_CUDA_VERSION} --extra-index-url {EXTRA_INDEX_URL}")