diff --git a/requirements.txt b/requirements.txt index fd8d538..37e8353 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ tqdm rotary_embedding_torch -transformers==4.31.0 +transformers>=4.43.0,<=4.43.3 tokenizers inflect progressbar diff --git a/setup.py b/setup.py index 13f0e27..8270349 100644 --- a/setup.py +++ b/setup.py @@ -27,9 +27,9 @@ setuptools.setup( 'unidecode', 'scipy', 'librosa', - 'transformers==4.31.0', - 'tokenizers==0.13.4.rc3', - 'scipy==1.13.1' + 'transformers>=4.43.0,<=4.43.3', + 'tokenizers', # ==0.13.4.rc3', + # 'scipy==1.13.1' # 'deepspeed==0.8.3', ], classifiers=[