aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_models.py
diff options
context:
space:
mode:
authorpapuSpartan <macabeg@icloud.com>2023-04-01 08:31:22 +0000
committerpapuSpartan <macabeg@icloud.com>2023-04-01 08:31:22 +0000
commit26ab018253cb078630fcdde47dbaee85f2466145 (patch)
tree9cad8cf3afe60ce281851144b250e7fa3e1be688 /modules/sd_models.py
parentef8c0440512f2fae9ceeb977b73f881c0afbb90a (diff)
downloadstable-diffusion-webui-gfx803-26ab018253cb078630fcdde47dbaee85f2466145.tar.gz
stable-diffusion-webui-gfx803-26ab018253cb078630fcdde47dbaee85f2466145.tar.bz2
stable-diffusion-webui-gfx803-26ab018253cb078630fcdde47dbaee85f2466145.zip
delay import
Diffstat (limited to 'modules/sd_models.py')
-rw-r--r--modules/sd_models.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py
index 0b74aa0f..2c05ec17 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -9,7 +9,6 @@ from omegaconf import OmegaConf
from os import mkdir
from urllib import request
import ldm.modules.midas as midas
-import tomesd
from ldm.util import instantiate_from_config
@@ -433,6 +432,7 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None, time_taken_
sd_model = instantiate_from_config(sd_config.model)
if shared.cmd_opts.token_merging:
+ import tomesd
ratio = shared.cmd_opts.token_merging_ratio
tomesd.apply_patch(sd_model, ratio=ratio)