Skip to content

Commit b8aebf4

Browse files
kaixuanliudg845
andauthored
add _repeated_blocks for ErnieImageTransformer2DModel (#13496)
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com>
1 parent b9d6420 commit b8aebf4

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

src/diffusers/models/transformers/transformer_ernie_image.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -290,6 +290,7 @@ def forward(self, x: torch.Tensor, conditioning: torch.Tensor) -> torch.Tensor:
290290

291291
class ErnieImageTransformer2DModel(ModelMixin, ConfigMixin):
292292
_supports_gradient_checkpointing = True
293+
_repeated_blocks = ["ErnieImageSharedAdaLNBlock"]
293294

294295
@register_to_config
295296
def __init__(

0 commit comments

Comments
 (0)