update.

ParagEkbote · ParagEkbote · commit 1f5373a828c6 · 2025-09-12T07:21:52.000Z
diff --git a/src/diffusers/models/attention_dispatch.py b/src/diffusers/models/attention_dispatch.py
@@ -774,7 +774,7 @@ def _flash_attention_hub(
         deterministic=deterministic,
         return_attn_probs=return_attn_probs,
     )
-    return out
+    return out(out[0], out[1]) if return_attn_probs else out
 
 
 @_AttentionBackendRegistry.register(

Original file line number	Diff line number	Diff line change
`@@ -774,7 +774,7 @@ def _flash_attention_hub(`
`774`	`774`	`deterministic=deterministic,`
`775`	`775`	`return_attn_probs=return_attn_probs,`
`776`	`776`	`)`
`777`		`- return out`
	`777`	`+ return out(out[0], out[1]) if return_attn_probs else out`
`778`	`778`
`779`	`779`
`780`	`780`	`@_AttentionBackendRegistry.register(`