mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
chore: pyflake lint fixes
This commit is contained in:
parent
50a266e064
commit
99475ab800
@ -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():
|
||||||
|
@ -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):
|
||||||
|
Loading…
Reference in New Issue
Block a user