From c66a53492c98222f2087de7af7e12da228d29534 Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Wed, 4 Dec 2024 20:33:25 -0600
Subject: [PATCH] forgot to add NTLK as a dependency, promoted sageattn as a
 default dependency since it works fine enough and seems agnostic

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 21d8fb4..4d255ce 100755
--- a/setup.py
+++ b/setup.py
@@ -73,6 +73,8 @@ setup(
 
         # for the web UI
         "gradio",
+        "nltk",
+        "sageattention==1.0.6",
     ],
     extras_require = {
         "all": [
@@ -86,7 +88,6 @@ setup(
 
             # attention helpers
             "xformers",
-            "sageattention==1.0.6",
             # "flash-attn" --no-build-isolation # commented out right now because I want to query this for Volta freaks like me who can't use it
             
             # other audio backend that doesn't prove fruitful