diff --git a/setup.py b/setup.py
index 5ecf932..4dd073a 100644
--- a/setup.py
+++ b/setup.py
@@ -22,6 +22,7 @@ setup(
         'peft',
         'deepspeed',
         'flash-attn==2.5.6',
-        'ir-datasets'
+        'ir-datasets',
+        'sentencepiece'
     ],
 )