diff --git a/src/diffusers/models/controlnets/controlnet_z_image.py b/src/diffusers/models/controlnets/controlnet_z_image.py index 85fa0d365547..f18233e91510 100644 --- a/src/diffusers/models/controlnets/controlnet_z_image.py +++ b/src/diffusers/models/controlnets/controlnet_z_image.py @@ -449,6 +449,7 @@ def __init__( qk_norm=True, ): super().__init__() + self.gradient_checkpointing = False self.control_layers_places = control_layers_places self.control_in_dim = control_in_dim self.control_refiner_layers_places = control_refiner_layers_places