diff --git a/setup.py b/setup.py index 68dc54393..86fea18b9 100644 --- a/setup.py +++ b/setup.py @@ -63,7 +63,7 @@ setuptools.setup( "statsmodels>=0.12.2", "psutil==5.8.0", "dataclasses", - "transformers[torch]==4.18", + "transformers[torch]", "datasets", "nltk", "rouge_score", @@ -93,7 +93,7 @@ setuptools.setup( "vowpalwabbit>=8.10.0, <9.0.0", ], "nlp": [ - "transformers[torch]==4.18", + "transformers[torch]==4.26", "datasets", "nltk", "rouge_score",