From 82d2a601c2d9079389ca2320caf13adb27de99e7 Mon Sep 17 00:00:00 2001 From: Steven Date: Tue, 18 Jul 2023 18:47:39 -0400 Subject: [PATCH] When searching for model info from the checkpoint specified from ckpt command line argument, strip the path from the argument so that we only search for the model's filename. --- modules/sd_models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index f81715d41..10c808047 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -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)