diff --git a/setup.py b/setup.py index 555580528f..197d632b43 100644 --- a/setup.py +++ b/setup.py @@ -31,7 +31,7 @@ "Pillow", "sacremoses", "torchvision", - "diffusers>=0.17.0", + "diffusers>=0.17.0,<0.32.0", "torchaudio", "einops", "timm", @@ -89,7 +89,7 @@ "executorch>=0.4.0", "transformers>=4.46", ], - "diffusers": ["diffusers"], + "diffusers": ["diffusers<0.32.0"], "intel": "optimum-intel>=1.18.0", "openvino": "optimum-intel[openvino]>=1.18.0", "nncf": "optimum-intel[nncf]>=1.18.0",