Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1746 from Phen-Ro/model_basename

Browse Source 
Search for models via the model file's basename
This commit is contained in:
Vladimir Mandic
2023-07-18 20:27:04 -04:00
committed by GitHub
parent debec28be6 82d2a601c2
commit 2035b99c82
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+2 -1
View File
@@ -606,7 +606,8 @@ def load_diffuser(checkpoint_info=None, already_loaded_state_dict=None, timer=No
try:
devices.set_cuda_params()
if shared.cmd_opts.ckpt is not None and model_data.initial: # initial load
model_name = modelloader.find_diffuser(shared.cmd_opts.ckpt)
ckpt_basename = os.path.basename(shared.cmd_opts.ckpt)
model_name = modelloader.find_diffuser(ckpt_basename)
if model_name is not None:
shared.log.info(f'Loading diffuser {op}: {model_name}')
model_file = modelloader.download_diffusers_model(hub_id=model_name)