We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0c8deaa commit 01f08d6Copy full SHA for 01f08d6
tox.ini
@@ -72,8 +72,9 @@ commands_pre =
72
pip install git+https://github.com/Dao-AILab/fast-hadamard-transform.git
73
74
# 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
+ # Skip triton because pytorch-triton is installed in the NGC PyTorch containers
+ pip install pip-mark-installed
77
+ pip install triton
78
pip install --no-build-isolation git+https://github.com/state-spaces/mamba.git
79
80
# Install Eagle-3 test dependencies
0 commit comments