Mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-06-07 21:20:49 +00:00
Code Issues 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 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)