fix: import only needed

This commit is contained in:
Bingsu
2023-05-18 07:51:05 +09:00
parent 209b703fbf
commit d6b5d2688f
8 changed files with 19 additions and 11 deletions

View File

@@ -1,4 +0,0 @@
from typing import TYPE_CHECKING
if not TYPE_CHECKING:
from modules import *

View File

@@ -59,4 +59,4 @@ if TYPE_CHECKING:
"""
else:
from modules.images import *
from modules.images import save_image

View File

@@ -11,4 +11,10 @@ if TYPE_CHECKING:
extensions_dir = os.path.join(os.path.dirname(__file__), "4")
extensions_builtin_dir = os.path.join(os.path.dirname(__file__), "5")
else:
from modules.paths import *
from modules.paths import (
data_path,
extensions_builtin_dir,
extensions_dir,
models_path,
script_path,
)

View File

@@ -163,4 +163,10 @@ if TYPE_CHECKING:
pass
else:
from modules.processing import *
from modules.processing import (
StableDiffusionProcessing,
StableDiffusionProcessingImg2Img,
StableDiffusionProcessingTxt2Img,
create_infotext,
process_images,
)

View File

@@ -7,4 +7,4 @@ if TYPE_CHECKING:
unsafe_torch_load = torch.load
else:
from modules.safe import *
from modules.safe import unsafe_torch_load

View File

@@ -10,4 +10,4 @@ if TYPE_CHECKING:
pass
else:
from modules.script_callbacks import *
from modules.script_callbacks import on_after_component, on_ui_settings

View File

@@ -78,4 +78,4 @@ if TYPE_CHECKING:
pass
else:
from modules.scripts import *
from modules.scripts import AlwaysVisible, PostprocessImageArgs, Script

View File

@@ -39,4 +39,4 @@ if TYPE_CHECKING:
cmd_opts = argparse.Namespace()
else:
from modules.shared import *
from modules.shared import OptionInfo, cmd_opts, opts