Fix triton dependency (#350)

casper-hansen · Feb 16, 2024 · 8849043 · 8849043
1 parent 969b290
commit 8849043
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -89,6 +89,7 @@ def get_kernels_whl_url(
  "transformers>=4.35.0",
  "tokenizers>=0.12.1",
  "typing_extensions>=4.8.0",
+ "triton",
  "accelerate",
  "datasets",
  "zstandard",