diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-12-06 15:16:10 +0800 |
|---|---|---|
| committer | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-12-06 15:16:10 +0800 |
| commit | 672dc4efa8e0da38426b121e7c7216d0a8e465fd (patch) | |
| tree | 87bf30352279ab71ef29d11e61661e1fe95c4e78 /modules | |
| parent | f5f89780ccf637979c0c88546080a13afd61c9a3 (diff) | |
Fix forced reload
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/sd_models.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index dcf816b3..d0046f88 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -801,7 +801,7 @@ def reload_model_weights(sd_model=None, info=None, forced_reload=False): if check_fp8(sd_model) != devices.fp8:
# load from state dict again to prevent extra numerical errors
forced_reload = True
- elif sd_model.sd_model_checkpoint == checkpoint_info.filename:
+ elif sd_model.sd_model_checkpoint == checkpoint_info.filename and not forced_reload:
return sd_model
sd_model = reuse_model_from_already_loaded(sd_model, checkpoint_info, timer)
|