aboutsummaryrefslogtreecommitdiff
path: root/modules/codeformer/codeformer_arch.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-05-10 08:25:25 +0300
committerAUTOMATIC <16777216c@gmail.com>2023-05-10 08:25:25 +0300
commit96d6ca4199e7c5eee8d451618de5161cea317c40 (patch)
tree8f101a345bcd1d66f4047b5e20918e2058e4dc7c /modules/codeformer/codeformer_arch.py
parent762265eab58cdb8f2d6398769bab43d8b8db0075 (diff)
manual fixes for ruff
Diffstat (limited to 'modules/codeformer/codeformer_arch.py')
-rw-r--r--modules/codeformer/codeformer_arch.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/codeformer/codeformer_arch.py b/modules/codeformer/codeformer_arch.py
index 11dcc3ee..f1a7cf09 100644
--- a/modules/codeformer/codeformer_arch.py
+++ b/modules/codeformer/codeformer_arch.py
@@ -7,7 +7,7 @@ from torch import nn, Tensor
import torch.nn.functional as F
from typing import Optional, List
-from modules.codeformer.vqgan_arch import *
+from modules.codeformer.vqgan_arch import VQAutoEncoder, ResBlock
from basicsr.utils import get_root_logger
from basicsr.utils.registry import ARCH_REGISTRY