hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 03:10:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

disable torch weight initialization and CLIP downloading/reading checkpoint to speedup creating sd model from config

Browse Source
This commit is contained in:
AUTOMATIC
2023-01-10 14:08:29 +03:00
parent ef75c98053
commit 0c3feb202c
2 changed files with 47 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modules/sd_models.py
View File

@@ -13,7 +13,7 @@ import ldm.modules.midas as midas
from ldm.util import instantiate_from_config
from modules import shared, modelloader, devices, script_callbacks, sd_vae
from modules import shared, modelloader, devices, script_callbacks, sd_vae, sd_disable_initialization
from modules.paths import models_path
from modules.sd_hijack_inpainting import do_inpainting_hijack, should_hijack_inpainting
@@ -319,7 +319,8 @@ def load_model(checkpoint_info=None):
if shared.cmd_opts.no_half:
sd_config.model.params.unet_config.params.use_fp16 = False
sd_model = instantiate_from_config(sd_config.model)
with sd_disable_initialization.DisableInitialization():
sd_model = instantiate_from_config(sd_config.model)
load_model_weights(sd_model, checkpoint_info)