Don't have triton as dependency

tridao · tridao · commit bee0ae23b9ea · 2024-12-06T14:09:47.000-08:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -19,7 +19,6 @@ dependencies = [
     "torch",
     "ninja",
     "einops",
-    "triton",
     "transformers",
     "packaging",
     "setuptools>=61.0.0",
diff --git a/setup.py b/setup.py
@@ -371,7 +371,7 @@ def run(self):
         "packaging",
         "ninja",
         "einops",
-        "triton",
+        # "triton",
         "transformers",
         # "causal_conv1d>=1.4.0",
     ],
