From 2f12246ec32a231a5f9bde18db1a69ed6e40c29e Mon Sep 17 00:00:00 2001
From: turboderp <11859846+turboderp@users.noreply.github.com>
Date: Mon, 7 Apr 2025 17:30:33 +0200
Subject: [PATCH] Fix requirements

---
 requirements.txt |  5 +++--
 setup.py         | 20 ++++++++------------
 2 files changed, 11 insertions(+), 14 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 5efaf80..d168bd4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,5 +3,6 @@ flash_attn>=2.7.4.post1
 tokenizers>=0.21.1
 numpy>=2.1.0
 rich
-tabulate
-typing_extensions
\ No newline at end of file
+typing_extensions
+safetensors>=0.3.2
+ninja
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 523da57..1ee4580 100644
--- a/setup.py
+++ b/setup.py
@@ -86,21 +86,17 @@ setup(
     license="MIT",
     author="turboderp",
     install_requires=[
-        "pandas",
-        "ninja",
-        "fastparquet",
-        "torch>=2.2.0",
-        "safetensors>=0.3.2",
-        "sentencepiece>=0.1.97",
-        "pygments",
-        "websockets",
-        "regex",
-        "numpy",
+        "torch>=2.6.0",
+        "flash_attn>=2.7.4.post1",
+        "tokenizers>=0.21.1",
+        "numpy>=2.1.0",
         "rich",
-        "pillow>=9.1.0"
+        "typing_extensions",
+        "ninja",
+        "safetensors>=0.3.2"
     ],
     include_package_data=True,
-    package_data={
+    package_data = {
         "": ["py.typed"],
     },
     verbose=verbose,