diff --git a/setup.py b/setup.py
index 0d86300..22ae20a 100644
--- a/setup.py
+++ b/setup.py
@@ -29,6 +29,7 @@ setuptools.setup(
         'librosa',
         'transformers==4.31.0',
         'tokenizers',
+        'deepspeed==0.8.3',
     ],
     classifiers=[
         "Programming Language :: Python :: 3",