Skip to content

Commit b519191

Browse files
Apply style fixes
1 parent 4ed9bcf commit b519191

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/diffusers/models/transformers/transformer_cogview4.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
from ..modeling_utils import ModelMixin
3131
from ..normalization import LayerNorm, RMSNorm
3232

33+
3334
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
3435

3536

@@ -585,9 +586,10 @@ def forward(self, hidden_states: torch.Tensor) -> Tuple[torch.Tensor, torch.Tens
585586

586587
class CogView4AdaLayerNormContinuous(nn.Module):
587588
"""
588-
CogView4-only final AdaLN: LN(x) -> Linear(cond) -> chunk -> affine.
589-
Matches Megatron: **no activation** before the Linear on conditioning embedding.
589+
CogView4-only final AdaLN: LN(x) -> Linear(cond) -> chunk -> affine. Matches Megatron: **no activation** before the
590+
Linear on conditioning embedding.
590591
"""
592+
591593
def __init__(
592594
self,
593595
embedding_dim: int,

0 commit comments

Comments
 (0)