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 0bc05f5 commit c40b4b2Copy full SHA for c40b4b2
pyproject.toml
@@ -41,15 +41,15 @@ dependencies = [
41
]
42
43
[project.optional-dependencies]
44
-dev = ["pre-commit>=3.0.4,<5.0"]
45
fp8 = ["llmcompressor"]
46
gptq = ["Cython", "gptqmodel>=1.7.3"]
47
mx = ["microxcaling>=1.1"]
48
-test = ["pillow"]
+opt = ["fms-model-optimizer[fp8, gptq, mx]"]
49
torchvision = ["torchvision>=0.17"]
50
-visualize = ["matplotlib", "graphviz", "pygraphviz"]
51
flash-attn = ["flash-attn>=2.5.3,<3.0"]
52
-opt = ["fms-model-optimizer[fp8, gptq]"]
+visualize = ["matplotlib", "graphviz", "pygraphviz"]
+dev = ["pre-commit>=3.0.4,<5.0"]
+test = ["pytest", "pillow"]
53
54
[project.urls]
55
homepage = "https://github.com/foundation-model-stack/fms-model-optimizer"
0 commit comments