Skip to content

Commit 6aa85f0

Browse files
committed
function name fix
1 parent 82320ad commit 6aa85f0

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/diffusers/models/attention_dispatch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1451,7 +1451,7 @@ def backward(
14511451
return grad_query, grad_key, grad_value, None, None, None, None, None, None, None, None, None
14521452

14531453

1454-
def TemplatedUnifiedAttention(
1454+
def _templated_unified_attention(
14551455
query: torch.Tensor,
14561456
key: torch.Tensor,
14571457
value: torch.Tensor,
@@ -1540,7 +1540,7 @@ def _templated_context_parallel_attention(
15401540
_parallel_config.context_parallel_config.ring_degree > 1
15411541
and _parallel_config.context_parallel_config.ulysses_degree > 1
15421542
):
1543-
return TemplatedUnifiedAttention(
1543+
return _templated_unified_attention(
15441544
query,
15451545
key,
15461546
value,

0 commit comments

Comments
 (0)