[BugFix] Rename attention_config to diffusion_attention_config (#3489)

Signed-off-by: gcanlin <canlinguosdu@gmail.com>
This commit is contained in:
Canlin Guo authored and GitHub committed 2026-05-11 11:27:35 +08:00
1 parent e1088026fa
commit 5e263b6929
5 files changed
+22 -38

No files matched your search

@@ -399,7 +399,7 @@ class SequenceParallelSplitHook(ModelHook):
if is_forward_context_available():
od_config = get_forward_context().omni_diffusion_config
if od_config is not None:
attention_config = od_config.attention_config
attention_config = od_config.diffusion_attention_config
attn_backend, _ = get_attn_backend_for_role(
role="self",