diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-09-28 22:31:53 +0100 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-09-30 07:37:05 +0300 |
| commit | f710ba11b99e25bb5149658502329e949e1cf6ae (patch) | |
| tree | 9456ae8e2ac4ba143823f70a3c8b2d5262900c55 | |
| parent | 642b7e333e2d87a8d3ef0bcb13c257347a9c13ed (diff) | |
use closest match checkpoint method
| -rw-r--r-- | scripts/xy_grid.py | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/scripts/xy_grid.py b/scripts/xy_grid.py index f8bc64c4..146663b0 100644 --- a/scripts/xy_grid.py +++ b/scripts/xy_grid.py @@ -45,11 +45,8 @@ def apply_sampler(p, x, xs): def apply_checkpoint(p, x, xs):
- applicable = [info for info in modules.sd_models.checkpoints_list.values() if x in info.title]
- assert len(applicable) > 0, f'Checkpoint {x} for found'
-
- info = applicable[0]
-
+ info = modules.sd_models.get_closet_checkpoint_match(x)
+ assert info is not None, f'Checkpoint for {x} not found'
modules.sd_models.reload_model_weights(shared.sd_model, info)
|