We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 28a2595 + d7f1ab5 commit 2f0b1a6Copy full SHA for 2f0b1a6
1 file changed
pyproject.toml
@@ -42,7 +42,7 @@ opt = ["fms-model-optimizer[fp8, gptq, mx]"]
42
aiu = ["ibm-fms>=0.0.8"]
43
torchvision = ["torchvision>=0.17"]
44
flash-attn = ["flash-attn>=2.5.3,<3.0"]
45
-triton = ["triton>=3.0,<3.4"]
+triton = ["triton>=3.0,<3.5"]
46
visualize = ["matplotlib", "graphviz", "pygraphviz", "tensorboard", "notebook"]
47
dev = ["pre-commit>=3.0.4,<5.0"]
48
test = ["pytest", "pillow"]
0 commit comments