StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

delay import

Browse Source
This commit is contained in:
papuSpartan 2023-04-01 03:31:22 -05:00
parent ef8c044051
commit 26ab018253
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_models.py
View File

@ -9,7 +9,6 @@ from omegaconf import OmegaConf
from os import mkdir from os import mkdir
from urllib import request from urllib import request
import ldm.modules.midas as midas import ldm.modules.midas as midas
import tomesd
from ldm.util import instantiate_from_config 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) sd_model = instantiate_from_config(sd_config.model)
if shared.cmd_opts.token_merging: if shared.cmd_opts.token_merging:
import tomesd
ratio = shared.cmd_opts.token_merging_ratio ratio = shared.cmd_opts.token_merging_ratio
tomesd.apply_patch(sd_model, ratio=ratio) tomesd.apply_patch(sd_model, ratio=ratio)