imports cleanup for ruff

This commit is contained in:
AUTOMATIC
2023-05-10 08:43:42 +03:00
parent 96d6ca4199
commit f741a98bac
48 changed files with 42 additions and 114 deletions

View File

@@ -1,14 +1,12 @@
# 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):