chore: pyflake lint fixes

This commit is contained in:
blessedcoolant 2023-08-29 05:16:23 +12:00
parent 50a266e064
commit 99475ab800
2 changed files with 5 additions and 4 deletions

View File

@ -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():

View File

@ -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):