from arch.hourglass.image_transformer_v2 import ImageTransformerDenoiserModelV2 from arch.swinir.swinir import SwinIR