AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Update transformer_flux.py (#9060)

Browse Source
This commit is contained in:
Frank (Haofan) Wang
2024-08-03 11:28:32 +08:00
committed by GitHub
parent b1f43d7189
commit a054c78495
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/diffusers/models/transformers/transformer_flux.py
View File

@@ -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)