diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2024-03-02 06:53:53 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2024-03-02 06:54:58 +0300 |
| commit | da67afe5f68497a04d1fd9173bbd256b73d9d251 (patch) | |
| tree | 2faaff7540bcfa0dec508df4d4ef567f131aa301 | |
| parent | 28bc85a20a282285710e17c4d86cf9db5e00d7db (diff) | |
call apply_alpha_schedule_override in load_model_weights for #14979
| -rw-r--r-- | modules/sd_models.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index fbd53adb..db72e120 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -428,6 +428,8 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer devices.dtype_unet = torch.float16
timer.record("apply half()")
+ apply_alpha_schedule_override(model)
+
for module in model.modules():
if hasattr(module, 'fp16_weight'):
del module.fp16_weight
@@ -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)
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
if sd_model is not None:
|