Skip to content

Commit 40714c9

Browse files
junqiangwuyiyixuxu
andauthored
Update src/diffusers/models/transformers/transformer_longcat_image.py
Co-authored-by: YiYi Xu <[email protected]>
1 parent d95fb05 commit 40714c9

File tree

1 file changed

+0
-12
lines changed

1 file changed

+0
-12
lines changed

src/diffusers/models/transformers/transformer_longcat_image.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -501,18 +501,6 @@ def forward(
501501
temb = self.time_embed( timestep, hidden_states.dtype )
502502
encoder_hidden_states = self.context_embedder(encoder_hidden_states)
503503

504-
if txt_ids.ndim == 3:
505-
logger.warning(
506-
"Passing `txt_ids` 3d torch.Tensor is deprecated."
507-
"Please remove the batch dimension and pass it as a 2d torch Tensor"
508-
)
509-
txt_ids = txt_ids[0]
510-
if img_ids.ndim == 3:
511-
logger.warning(
512-
"Passing `img_ids` 3d torch.Tensor is deprecated."
513-
"Please remove the batch dimension and pass it as a 2d torch Tensor"
514-
)
515-
img_ids = img_ids[0]
516504

517505
ids = torch.cat((txt_ids, img_ids), dim=0)
518506
if is_torch_npu_available():

0 commit comments

Comments
 (0)