michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

return shortest checkpoint title match

Browse source
This commit is contained in:
DepFA 2022-09-29 19:08:03 +01:00 committed by AUTOMATIC1111
parent f710ba11b9
commit ebd2c48115
1 changed files with 1 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
modules/sd_models.py
View file

@ -62,19 +62,9 @@ def list_models():
checkpoints_list[title] = CheckpointInfo(filename, title, h, model_name)
def get_closet_checkpoint_match(searchString):
checkpointValues = checkpoints_list.values()
applicable = [info for info in checkpointValues if searchString.upper() == ''.join(info.title.rpartition('.ckpt')[0]).upper()]
if len(applicable) == 0:
applicable = [info for info in checkpointValues if searchString.upper() == ''.join(info.title.rpartition('.ckpt')[:2]).upper()]
if len(applicable) == 0:
applicable = [info for info in checkpointValues if searchString in info.title]
applicable = sorted([info for info in checkpoints_list.values() if searchString in info.title], key = lambda x:len(x.title))
if len(applicable)>0:
return applicable[0]
return None
def model_hash(filename):