diff --git a/sd_webui/__init__.py b/sd_webui/__init__.py index 93c742f..e69de29 100644 --- a/sd_webui/__init__.py +++ b/sd_webui/__init__.py @@ -1,4 +0,0 @@ -from typing import TYPE_CHECKING - -if not TYPE_CHECKING: - from modules import * diff --git a/sd_webui/images.py b/sd_webui/images.py index 8a2cb8d..b4a2dbc 100644 --- a/sd_webui/images.py +++ b/sd_webui/images.py @@ -59,4 +59,4 @@ if TYPE_CHECKING: """ else: - from modules.images import * + from modules.images import save_image diff --git a/sd_webui/paths.py b/sd_webui/paths.py index 688f251..159af00 100644 --- a/sd_webui/paths.py +++ b/sd_webui/paths.py @@ -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, + ) diff --git a/sd_webui/processing.py b/sd_webui/processing.py index 490b76d..fb9999c 100644 --- a/sd_webui/processing.py +++ b/sd_webui/processing.py @@ -163,4 +163,10 @@ if TYPE_CHECKING: pass else: - from modules.processing import * + from modules.processing import ( + StableDiffusionProcessing, + StableDiffusionProcessingImg2Img, + StableDiffusionProcessingTxt2Img, + create_infotext, + process_images, + ) diff --git a/sd_webui/safe.py b/sd_webui/safe.py index 92c5acf..c17a1b9 100644 --- a/sd_webui/safe.py +++ b/sd_webui/safe.py @@ -7,4 +7,4 @@ if TYPE_CHECKING: unsafe_torch_load = torch.load else: - from modules.safe import * + from modules.safe import unsafe_torch_load diff --git a/sd_webui/script_callbacks.py b/sd_webui/script_callbacks.py index 959c8e8..b1c9136 100644 --- a/sd_webui/script_callbacks.py +++ b/sd_webui/script_callbacks.py @@ -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 diff --git a/sd_webui/scripts.py b/sd_webui/scripts.py index 0f41309..2872a7f 100644 --- a/sd_webui/scripts.py +++ b/sd_webui/scripts.py @@ -78,4 +78,4 @@ if TYPE_CHECKING: pass else: - from modules.scripts import * + from modules.scripts import AlwaysVisible, PostprocessImageArgs, Script diff --git a/sd_webui/shared.py b/sd_webui/shared.py index 2d28598..e71260a 100644 --- a/sd_webui/shared.py +++ b/sd_webui/shared.py @@ -39,4 +39,4 @@ if TYPE_CHECKING: cmd_opts = argparse.Namespace() else: - from modules.shared import * + from modules.shared import OptionInfo, cmd_opts, opts