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): 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 # TODO: get an explicit interface for this in diffusers: https://github.com/huggingface/diffusers/issues/556
for m in model.modules(): for m in model.modules():

View File

@ -1,10 +1,10 @@
from __future__ import annotations from __future__ import annotations
from contextlib import contextmanager from contextlib import contextmanager
from typing import Union, List from typing import List, Union
import diffusers
import torch.nn as nn 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): def _conv_forward_asymmetric(self, input, weight, bias):