From 6509bf7e89b1aca0b674b6715920c131479c52d5 Mon Sep 17 00:00:00 2001 From: yunjoochoi Date: Sun, 15 Mar 2026 12:26:47 +0900 Subject: [PATCH] Fix missing gradient_checkpointing init in ZImageControlNetModel --- src/diffusers/models/controlnets/controlnet_z_image.py | 1 + 1 file changed, 1 insertion(+) 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