Update transformer_flux.py (#9060)

haofanwang · web-flow · commit a054c7849550 · 2024-08-03T08:58:32.000+05:30
diff --git a/src/diffusers/models/transformers/transformer_flux.py b/src/diffusers/models/transformers/transformer_flux.py
@@ -373,7 +373,6 @@ def forward(
         )
         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)
 

Original file line number	Diff line number	Diff line change
`@@ -373,7 +373,6 @@ def forward(`
`373`	`373`	`)`
`374`	`374`	`encoder_hidden_states = self.context_embedder(encoder_hidden_states)`
`375`	`375`
`376`		`- print(f"{txt_ids.shape=}, {img_ids.shape=}")`
`377`	`376`	`ids = torch.cat((txt_ids, img_ids), dim=1)`
`378`	`377`	`image_rotary_emb = self.pos_embed(ids)`
`379`	`378`