imports cleanup for ruff

import glob

import os

import re

import torch

from modules import devices, modelloader

from scunet_model_arch import SCUNet as net

from modules.shared import opts

from modules import images

class UpscalerScuNET(modules.upscaler.Upscaler):

import contextlib

import os

import numpy as np

from tqdm import tqdm

from modules import modelloader, devices, script_callbacks, shared

from modules.shared import cmd_opts, opts, state

from modules.shared import opts, state

from swinir_model_arch import SwinIR as net

from swinir_model_arch_v2 import Swin2SR as net2

from modules.upscaler import Upscaler, UpscalerData

# this file is copied from CodeFormer repository. Please see comment in modules/codeformer_model.py

import math

import numpy as np

import torch

from torch import nn, Tensor

import torch.nn.functional as F

from typing import Optional, List

from typing import Optional

from modules.codeformer.vqgan_arch import VQAutoEncoder, ResBlock

from basicsr.utils import get_root_logger

from basicsr.utils.registry import ARCH_REGISTRY

def calc_mean_std(feat, eps=1e-5):

https://github.com/samb-t/unleashing-transformers/blob/master/models/vqgan.py

'''

import numpy as np

import torch

import torch.nn as nn

import torch.nn.functional as F

import copy

from basicsr.utils import get_root_logger

from basicsr.utils.registry import ARCH_REGISTRY