From 8849043a0d2661ff178df6d8028114c1980328e3 Mon Sep 17 00:00:00 2001
From: Casper <casperbh.96@gmail.com>
Date: Fri, 16 Feb 2024 09:41:22 +0100
Subject: [PATCH] Fix triton dependency (#350)

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index f8b18b84..63641124 100644
--- 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",