We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ef47726 commit 5a2e0f7Copy full SHA for 5a2e0f7
src/diffusers/models/transformers/transformer_hidream_image.py
@@ -918,5 +918,5 @@ def forward(
918
unscale_lora_layers(self, lora_scale)
919
920
if not return_dict:
921
- return (output, hidden_states_masks)
922
- return Transformer2DModelOutput(sample=output, mask=hidden_states_masks)
+ return (output,)
+ return Transformer2DModelOutput(sample=output)
0 commit comments