diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-17 13:49:36 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-17 13:49:36 +0300 |
| commit | 304222ef94d1c3c60fab466a96c448868f391bce (patch) | |
| tree | 7b5aa24ac29865f3c4c14eb755ef3f5d6d7421c3 /modules/sd_models.py | |
| parent | 99585b3514e2d7e987651d5c6a0806f933af012b (diff) | |
X/Y plot support for switching checkpoints.
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 036af0e4..4bd70fc5 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -127,9 +127,9 @@ def load_model(): return sd_model
-def reload_model_weights(sd_model):
+def reload_model_weights(sd_model, info=None):
from modules import lowvram, devices
- checkpoint_info = select_checkpoint()
+ checkpoint_info = info or select_checkpoint()
if sd_model.sd_model_checkpint == checkpoint_info.filename:
return
|