Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix lint

Browse Source
This commit is contained in:
suluyan.sly@alibaba-inc.com
2024-12-09 20:17:00 +08:00
parent 5bcf1a485e
commit 6805e01d3c
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modelscope/models/cv/image_super_resolution_pasd/unet_2d_blocks.py
View File

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

3
modelscope/models/cv/image_super_resolution_pasd_v2/unet_2d_blocks.py
View File

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