Fix mamba triton install

Signed-off-by: Keval Morabia <[email protected]>

Author: kevalmorabia97

tox.ini

@@ -74,7 +74,7 @@ commands_pre =
     # Install Mamba model dependencies (takes 8-10mins!)
     # Skip triton because pytorch-triton is installed in the NGC PyTorch containers
     pip install pip-mark-installed
-    pip install triton
+    pip-mark-installed triton
     pip install --no-build-isolation git+https://github.com/state-spaces/mamba.git
 
     # Install Eagle-3 test dependencies