Skip to content

Commit 6856ee6

Browse files
committed
Add _repeated_blocks attribute to SkyReelsV2Transformer3DModel
1 parent f09be25 commit 6856ee6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/diffusers/models/transformers/transformer_skyreels_v2.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -529,6 +529,7 @@ class SkyReelsV2Transformer3DModel(
529529
_no_split_modules = ["SkyReelsV2TransformerBlock"]
530530
_keep_in_fp32_modules = ["time_embedder", "scale_shift_table", "norm1", "norm2", "norm3"]
531531
_keys_to_ignore_on_load_unexpected = ["norm_added_q"]
532+
_repeated_blocks = ["SkyReelsV2TransformerBlock"]
532533

533534
@register_to_config
534535
def __init__(

0 commit comments

Comments
 (0)