Skip to content

GitLab

Unverified Commit 01e35551 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

Enable regional compilation on z-image transformer model (#12736) 

up
parent 6bf668c4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/models/transformers/transformer_z_image.py
View file @ 01e35551
...@@ -328,6 +328,7 @@ class RopeEmbedder: ...@@ -328,6 +328,7 @@ class RopeEmbedder:
class ZImageTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin, FromOriginalModelMixin): class ZImageTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin, FromOriginalModelMixin):
_supports_gradient_checkpointing = True _supports_gradient_checkpointing = True
_no_split_modules = ["ZImageTransformerBlock"] _no_split_modules = ["ZImageTransformerBlock"]
_repeated_blocks = ["ZImageTransformerBlock"]
_skip_layerwise_casting_patterns = ["t_embedder", "cap_embedder"] # precision sensitive layers _skip_layerwise_casting_patterns = ["t_embedder", "cap_embedder"] # precision sensitive layers
@register_to_config @register_to_config
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment