Skip to content

Commit 2a82064

Browse files
committed
make style
1 parent 174621f commit 2a82064

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

src/diffusers/models/normalization.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,7 @@
2222

2323
from ..utils import is_torch_version
2424
from .activations import get_activation
25-
from .embeddings import (
26-
CombinedTimestepLabelEmbeddings,
27-
PixArtAlphaCombinedTimestepSizeEmbeddings
28-
)
25+
from .embeddings import CombinedTimestepLabelEmbeddings, PixArtAlphaCombinedTimestepSizeEmbeddings
2926

3027

3128
class AdaLayerNorm(nn.Module):

src/diffusers/models/transformers/transformer_allegro.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@
3737
class AllegroTransformerBlock(nn.Module):
3838
r"""
3939
Transformer block used in [Allegro](https://github.com/rhymes-ai/Allegro) model.
40+
4041
Args:
4142
dim (`int`):
4243
The number of channels in the input and output.
@@ -176,6 +177,7 @@ class AllegroTransformer3DModel(ModelMixin, ConfigMixin):
176177

177178
"""
178179
A 3D Transformer model for video-like data.
180+
179181
Args:
180182
patch_size (`int`, defaults to `2`):
181183
The size of spatial patches to use in the patch embedding layer.

0 commit comments

Comments
 (0)