Update hidiffusion.py

fix diffusers 0.27.2 bug
2024-08-26 13:20:47 +08:00
parent 011b32bdf0
commit 8463f714c7
1 changed files with 1 additions and 1 deletions
@@ -16,7 +16,7 @@ import warnings

 dif_version = str(diffusers.__version__)
 dif_version_int= int(dif_version.split(".")[1])
-if dif_version_int>27:
+if dif_version_int>=27:
    from diffusers.models.unets.unet_2d_condition import UNet2DConditionOutput
    old_diffusers = False
 else: