From fec56de319b7a45dbeedec3a166175a4154b57f1 Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Sun, 11 Aug 2024 15:58:30 -0700 Subject: [PATCH] show xformers on UI --- modules/shared.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 = {}