diff --git a/setup.py b/setup.py index dab9972b8b..81b3d74c9c 100644 --- a/setup.py +++ b/setup.py @@ -72,6 +72,7 @@ "timm", "h5py", "numpy<1.24.0", + "transformers[sentencepiece]>=4.26.0,<4.38.0", ], "diffusers": ["diffusers"], "intel": "optimum-intel>=1.15.0",