Skip to content

Commit 9051272

Browse files
committed
x_pad_token
1 parent 8688fa6 commit 9051272

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/controlnets/controlnet_z_image.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -409,7 +409,7 @@ def forward(
409409

410410
# Match t_embedder output dtype to control_context for layerwise casting compatibility
411411
adaln_input = t.type_as(control_context)
412-
control_context[torch.cat(x_inner_pad_mask)] = self.x_pad_token
412+
control_context[torch.cat(x_inner_pad_mask)] = transformer.x_pad_token
413413
control_context = list(control_context.split(x_item_seqlens, dim=0))
414414
x_freqs_cis = list(transformer.rope_embedder(torch.cat(x_pos_ids, dim=0)).split(x_item_seqlens, dim=0))
415415

0 commit comments

Comments
 (0)