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

Fix model caching requiring deepcopy

Browse Source
This commit is contained in:
Muhammad Rizqi Nur 2022-11-17 18:04:10 +07:00
parent c7be83bf02
commit 9fdc343dca
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/sd_vae.py
View File

@ -4,6 +4,7 @@ from collections import namedtuple
from modules import shared, devices, script_callbacks from modules import shared, devices, script_callbacks
from modules.paths import models_path from modules.paths import models_path
import glob import glob
from copy import deepcopy
model_dir = "Stable-diffusion" model_dir = "Stable-diffusion"
@ -40,7 +41,7 @@ def store_base_vae(model):
global base_vae, checkpoint_info global base_vae, checkpoint_info
if checkpoint_info != model.sd_checkpoint_info: if checkpoint_info != model.sd_checkpoint_info:
assert not loaded_vae_file, "Trying to store non-base VAE!" assert not loaded_vae_file, "Trying to store non-base VAE!"
base_vae = model.first_stage_model.state_dict().copy() base_vae = deepcopy(model.first_stage_model.state_dict())
checkpoint_info = model.sd_checkpoint_info checkpoint_info = model.sd_checkpoint_info