From 6805e01d3c8ea567f45846740cdf8a8fd785aab0 Mon Sep 17 00:00:00 2001 From: "suluyan.sly@alibaba-inc.com" Date: Mon, 9 Dec 2024 20:17:00 +0800 Subject: [PATCH] fix lint --- .../models/cv/image_super_resolution_pasd/unet_2d_blocks.py | 3 ++- .../models/cv/image_super_resolution_pasd_v2/unet_2d_blocks.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/modelscope/models/cv/image_super_resolution_pasd/unet_2d_blocks.py b/modelscope/models/cv/image_super_resolution_pasd/unet_2d_blocks.py index 67fec789..a36316e4 100644 --- a/modelscope/models/cv/image_super_resolution_pasd/unet_2d_blocks.py +++ b/modelscope/models/cv/image_super_resolution_pasd/unet_2d_blocks.py @@ -8,10 +8,11 @@ import torch.nn.functional as F from diffusers.models.attention_processor import (Attention, AttnAddedKVProcessor, AttnAddedKVProcessor2_0) -from diffusers.models.transformers.dual_transformer_2d import DualTransformer2DModel from diffusers.models.resnet import (Downsample2D, FirDownsample2D, FirUpsample2D, KDownsample2D, KUpsample2D, ResnetBlock2D, Upsample2D) +from diffusers.models.transformers.dual_transformer_2d import \ + DualTransformer2DModel from torch import nn from .attention import AdaGroupNorm diff --git a/modelscope/models/cv/image_super_resolution_pasd_v2/unet_2d_blocks.py b/modelscope/models/cv/image_super_resolution_pasd_v2/unet_2d_blocks.py index b76467b2..8704cceb 100644 --- a/modelscope/models/cv/image_super_resolution_pasd_v2/unet_2d_blocks.py +++ b/modelscope/models/cv/image_super_resolution_pasd_v2/unet_2d_blocks.py @@ -20,11 +20,12 @@ from diffusers.models.activations import get_activation from diffusers.models.attention_processor import (Attention, AttnAddedKVProcessor, AttnAddedKVProcessor2_0) -from diffusers.models.transformers.dual_transformer_2d import DualTransformer2DModel from diffusers.models.normalization import AdaLayerNorm from diffusers.models.resnet import (Downsample2D, FirDownsample2D, FirUpsample2D, KDownsample2D, KUpsample2D, ResnetBlock2D, Upsample2D) +from diffusers.models.transformers.dual_transformer_2d import \ + DualTransformer2DModel from diffusers.models.transformers.transformer_2d import Transformer2DModel from diffusers.utils import is_torch_version, logging from einops import rearrange