Skip to content

Commit 9854807

Browse files
committed
make style.
1 parent 00e1209 commit 9854807

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414
import inspect
15-
from typing import List, Optional, Tuple, Union, Dict, Callable
15+
from typing import Callable, Dict, List, Optional, Tuple, Union
1616

1717
import torch
1818
from transformers import T5Tokenizer, UMT5EncoderModel
@@ -24,7 +24,6 @@
2424
from ...utils import is_torch_xla_available, logging, replace_example_docstring
2525
from ...utils.torch_utils import randn_tensor
2626
from ..pipeline_utils import DiffusionPipeline, ImagePipelineOutput
27-
from ...callbacks import MultiPipelineCallbacks, PipelineCallback
2827

2928

3029
if is_torch_xla_available():
@@ -172,7 +171,7 @@ def check_inputs(
172171
raise ValueError(
173172
f"`height` and `width` have to be divisible by {self.vae_scale_factor * 2} but are {height} and {width}."
174173
)
175-
174+
176175
if callback_on_step_end_tensor_inputs is not None and not all(
177176
k in self._callback_tensor_inputs for k in callback_on_step_end_tensor_inputs
178177
):

src/diffusers/pipelines/lumina/pipeline_lumina.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
import math
1818
import re
1919
import urllib.parse as ul
20-
from typing import List, Optional, Tuple, Union, Dict, Callable
20+
from typing import Callable, Dict, List, Optional, Tuple, Union
2121

2222
import torch
2323
from transformers import AutoModel, AutoTokenizer
@@ -37,7 +37,6 @@
3737
)
3838
from ...utils.torch_utils import randn_tensor
3939
from ..pipeline_utils import DiffusionPipeline, ImagePipelineOutput
40-
from ...callbacks import MultiPipelineCallbacks, PipelineCallback
4140

4241

4342
if is_torch_xla_available():
@@ -409,7 +408,7 @@ def check_inputs(
409408
raise ValueError(
410409
f"`height` and `width` have to be divisible by {self.vae_scale_factor * 2} but are {height} and {width}."
411410
)
412-
411+
413412
if callback_on_step_end_tensor_inputs is not None and not all(
414413
k in self._callback_tensor_inputs for k in callback_on_step_end_tensor_inputs
415414
):

0 commit comments

Comments
 (0)