Skip to content

GitLab

Unverified Commit a054c784 authored by Frank (Haofan) Wang's avatar Frank (Haofan) Wang Committed by GitHub
Browse files

Update transformer_flux.py (#9060)

parent b1f43d71
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
src/diffusers/models/transformers/transformer_flux.py
View file @ a054c784
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment