[BugFix] Rename attention_config to diffusion_attention_config (#3489)
Signed-off-by: gcanlin <canlinguosdu@gmail.com>
This commit is contained in:
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",
|
||||
|
||||
Reference in new issue
Block a user