diff --git a/modules/shared.py b/modules/shared.py index af5ec283..564f613e 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -7,6 +7,7 @@ from modules import shared_cmd_options, shared_gradio_themes, options, shared_it from modules.paths_internal import models_path, script_path, data_path, sd_configs_path, sd_default_config, sd_model_file, default_sd_model_file, extensions_dir, extensions_builtin_dir # noqa: F401 from modules import util from typing import TYPE_CHECKING +from backend import memory_management if TYPE_CHECKING: from modules import shared_state, styles, interrogate, shared_total_tqdm, memmon @@ -26,7 +27,7 @@ device: str = None weight_load_location: str = None -xformers_available = False +xformers_available = memory_management.xformers_enabled() hypernetworks = {}