Skip to content

Commit 570558b

Browse files
committed
style
1 parent 5f560ca commit 570558b

File tree

3 files changed

+4
-6
lines changed

3 files changed

+4
-6
lines changed

src/diffusers/models/controlnets/controlnet_sd3.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -56,9 +56,7 @@ def __init__(
5656
out_channels: int = 16,
5757
pos_embed_max_size: int = 96,
5858
extra_conditioning_channels: int = 0,
59-
dual_attention_layers: Tuple[
60-
int, ...
61-
] = (),
59+
dual_attention_layers: Tuple[int, ...] = (),
6260
qk_norm: Optional[str] = None,
6361
):
6462
super().__init__()

src/diffusers/models/transformers/transformer_sd3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,9 @@
1515

1616
from typing import Any, Dict, List, Optional, Tuple, Union
1717

18+
import numpy as np
1819
import torch
1920
import torch.nn as nn
20-
import numpy as np
2121

2222
from ...configuration_utils import ConfigMixin, register_to_config
2323
from ...loaders import FromOriginalModelMixin, PeftAdapterMixin

tests/pipelines/controlnet_sd3/test_controlnet_sd3.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,11 @@
1515

1616
import gc
1717
import unittest
18+
from typing import Optional
1819

1920
import numpy as np
2021
import pytest
2122
import torch
22-
from typing import Optional
2323
from transformers import AutoTokenizer, CLIPTextConfig, CLIPTextModelWithProjection, CLIPTokenizer, T5EncoderModel
2424

2525
from diffusers import (
@@ -59,7 +59,7 @@ class StableDiffusion3ControlNetPipelineFastTests(unittest.TestCase, PipelineTes
5959
]
6060
)
6161
batch_params = frozenset(["prompt", "negative_prompt"])
62-
62+
6363
def get_dummy_components(self, num_controlnet_layers: int = 3, qk_norm: Optional[str] = "rms_norm"):
6464
torch.manual_seed(0)
6565
transformer = SD3Transformer2DModel(

0 commit comments

Comments
 (0)