diff --git a/requirements.txt b/requirements.txt index 1289318b..79336152 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ coremltools>=8.0 diffusers[torch]==0.30.2 diffusionkit==0.4.0 torch -transformers==4.44.2 +transformers==5.3.0 scipy scikit-learn pytest diff --git a/setup.py b/setup.py index 644fa456..23cbe105 100644 --- a/setup.py +++ b/setup.py @@ -17,7 +17,7 @@ "coremltools>=8.0", "diffusers[torch]==0.30.2", "torch", - "transformers==4.44.2", + "transformers==5.3.0", "huggingface-hub==0.24.6", "scipy", "numpy<1.24",