Commit f710ba11 authored by DepFA's avatar DepFA Committed by AUTOMATIC1111
Browse files

use closest match checkpoint method

parent 642b7e33
Loading
Loading
Loading
Loading
+2 −5
Original line number Diff line number Diff line
@@ -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)