diff --git a/setup.py b/setup.py index 484998a..ec96608 100644 --- a/setup.py +++ b/setup.py @@ -13,7 +13,8 @@ 'numpy', 'transformers', 'bitsandbytes', - 'accelerate' + 'accelerate', + 'sentencepiece' ], classifiers=[