From 618de486d2701fd03dbcb71868505e5c75c372d7 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Thu, 27 Nov 2025 16:28:33 +0530 Subject: [PATCH] up --- src/diffusers/models/transformers/transformer_z_image.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/models/transformers/transformer_z_image.py b/src/diffusers/models/transformers/transformer_z_image.py index 3ad835ceeeb0..4f2d56ea8f4d 100644 --- a/src/diffusers/models/transformers/transformer_z_image.py +++ b/src/diffusers/models/transformers/transformer_z_image.py @@ -328,6 +328,7 @@ def __call__(self, ids: torch.Tensor): class ZImageTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin, FromOriginalModelMixin): _supports_gradient_checkpointing = True _no_split_modules = ["ZImageTransformerBlock"] + _repeated_blocks = ["ZImageTransformerBlock"] _skip_layerwise_casting_patterns = ["t_embedder", "cap_embedder"] # precision sensitive layers @register_to_config