From 99475ab8003459468aaa5f3994ee6fec522bf2d0 Mon Sep 17 00:00:00 2001 From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com> Date: Tue, 29 Aug 2023 05:16:23 +1200 Subject: [PATCH] chore: pyflake lint fixes --- invokeai/backend/image_util/seamless.py | 3 ++- invokeai/backend/model_management/seamless.py | 6 +++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/invokeai/backend/image_util/seamless.py b/invokeai/backend/image_util/seamless.py index 6fb2617901..8a2580bfcc 100644 --- a/invokeai/backend/image_util/seamless.py +++ b/invokeai/backend/image_util/seamless.py @@ -20,7 +20,8 @@ def _conv_forward_asymmetric(self, input, weight, bias): def configure_model_padding(model, seamless, seamless_axes): """ - Modifies the 2D convolution layers to use a circular padding mode based on the `seamless` and `seamless_axes` options. + Modifies the 2D convolution layers to use a circular padding mode based on + the `seamless` and `seamless_axes` options. """ # TODO: get an explicit interface for this in diffusers: https://github.com/huggingface/diffusers/issues/556 for m in model.modules(): diff --git a/invokeai/backend/model_management/seamless.py b/invokeai/backend/model_management/seamless.py index 4e9587a86f..997ff3563f 100644 --- a/invokeai/backend/model_management/seamless.py +++ b/invokeai/backend/model_management/seamless.py @@ -1,10 +1,10 @@ from __future__ import annotations from contextlib import contextmanager -from typing import Union, List -import diffusers +from typing import List, Union + import torch.nn as nn -from diffusers.models import UNet2DModel, AutoencoderKL +from diffusers.models import AutoencoderKL, UNet2DModel def _conv_forward_asymmetric(self, input, weight, bias):