Add sentencepiece dependency

probcomp · Sep 12, 2023 · c6b2eb2 · c6b2eb2
1 parent d25e2af
commit c6b2eb2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -13,7 +13,8 @@
                       'numpy',
                       'transformers',
                       'bitsandbytes',
-                      'accelerate'
+                      'accelerate',
+                      'sentencepiece'
                       ],
 
     classifiers=[