depenency fix

artitw · Oct 26, 2024 · 44d8fca · 44d8fca
1 parent 192638d
commit 44d8fca
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
   name="text2text",
-  version="1.8.0",
+  version="1.8.1",
   author="artitw",
   author_email="artitw@gmail.com",
   description="Text2Text: Crosslingual NLP/G toolkit",
@@ -35,9 +35,8 @@
     'torch',
     'tqdm',
     'transformers',
-    'trl',
     'peft',
     'bitsandbytes',
-    'flash-attn'
+    'trl'
   ],
 )