From a054c784955023385cf91e792db10970d00ca281 Mon Sep 17 00:00:00 2001 From: "Frank (Haofan) Wang" Date: Sat, 3 Aug 2024 11:28:32 +0800 Subject: [PATCH] Update transformer_flux.py (#9060) --- src/diffusers/models/transformers/transformer_flux.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/diffusers/models/transformers/transformer_flux.py b/src/diffusers/models/transformers/transformer_flux.py index 364867275d..391ca1418d 100644 --- a/src/diffusers/models/transformers/transformer_flux.py +++ b/src/diffusers/models/transformers/transformer_flux.py @@ -373,7 +373,6 @@ class FluxTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin): ) encoder_hidden_states = self.context_embedder(encoder_hidden_states) - print(f"{txt_ids.shape=}, {img_ids.shape=}") ids = torch.cat((txt_ids, img_ids), dim=1) image_rotary_emb = self.pos_embed(ids)