diff --git a/requirements.txt b/requirements.txt
index c9ffe3e..ed07500 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,6 @@
 pandas
+ninja
 fastparquet
 torch>=2.0.1
 safetensors>=0.3.2
-sentencepiece>=0.1.97
\ No newline at end of file
+sentencepiece>=0.1.97
diff --git a/setup.py b/setup.py
index 42f150b..11c12fd 100644
--- a/setup.py
+++ b/setup.py
@@ -11,6 +11,7 @@ setup(
     author = "bb",
     install_requires = [
         "pandas",
+        "ninja",
         "fastparquet",
         "torch>=2.0.1",
         "safetensors>=0.3.2",