Skip to content

S
Stable Diffusion Webui
You need to sign in or sign up before continuing.
Commit 1a51b166 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files
Options

call apply_alpha_schedule_override in load_model_weights for #14979

parent 06b9200e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
modules/sd_models.py
View file @ 1a51b166
...@@ -428,6 +428,8 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer ...@@ -428,6 +428,8 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer
devices.dtype_unet = torch.float16 devices.dtype_unet = torch.float16
timer.record("apply half()") timer.record("apply half()")
apply_alpha_schedule_override(model)
for module in model.modules(): for module in model.modules():
if hasattr(module, 'fp16_weight'): if hasattr(module, 'fp16_weight'):
del module.fp16_weight del module.fp16_weight
...@@ -843,7 +845,6 @@ def reload_model_weights(sd_model=None, info=None, forced_reload=False): ...@@ -843,7 +845,6 @@ def reload_model_weights(sd_model=None, info=None, forced_reload=False):
sd_model = reuse_model_from_already_loaded(sd_model, checkpoint_info, timer) sd_model = reuse_model_from_already_loaded(sd_model, checkpoint_info, timer)
if not forced_reload and sd_model is not None and sd_model.sd_checkpoint_info.filename == checkpoint_info.filename: if not forced_reload and sd_model is not None and sd_model.sd_checkpoint_info.filename == checkpoint_info.filename:
apply_alpha_schedule_override(sd_model)
return sd_model return sd_model
if sd_model is not None: if sd_model is not None:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment