Skip to content

Commit 01f08d6

Browse files
Fix mamba cicd installation
Signed-off-by: Keval Morabia <[email protected]>
1 parent 0c8deaa commit 01f08d6

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tox.ini

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,9 @@ commands_pre =
7272
pip install git+https://github.com/Dao-AILab/fast-hadamard-transform.git
7373

7474
# Install Mamba model dependencies (takes 8-10mins!)
75-
# Install same triton as pytorch-triton in the NGC PyTorch 25.08 docker otherwise Mamba may install an incompatible version
76-
pip install triton==3.3.1
75+
# Skip triton because pytorch-triton is installed in the NGC PyTorch containers
76+
pip install pip-mark-installed
77+
pip install triton
7778
pip install --no-build-isolation git+https://github.com/state-spaces/mamba.git
7879

7980
# Install Eagle-3 test dependencies

0 commit comments

Comments
 (0)