add deepspeed to setup.py

This commit is contained in:
manmay nakhashi 2023-09-24 20:09:12 +05:30 committed by GitHub
parent 518e73b645
commit 40e2db6d4e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -29,6 +29,7 @@ setuptools.setup(
'librosa',
'transformers==4.31.0',
'tokenizers',
'deepspeed==0.8.3',
],
classifiers=[
"Programming Language :: Python :: 3",