Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "a980ef2f08f8d3b40beddefb9dcce752e31d3295"
Unverified Commit 5a2e0f71 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

update output for Hidream transformer (#11366) 

up
parent ef47726e
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/diffusers/models/transformers/transformer_hidream_image.py
View file @ 5a2e0f71
...@@ -918,5 +918,5 @@ class HiDreamImageTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin): ...@@ -918,5 +918,5 @@ class HiDreamImageTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin):
unscale_lora_layers(self, lora_scale) unscale_lora_layers(self, lora_scale)
if not return_dict: if not return_dict:
return (output, hidden_states_masks) return (output,)
return Transformer2DModelOutput(sample=output, mask=hidden_states_masks) return Transformer2DModelOutput(sample=output)
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