From 96a900d1fe730ee44748d7a0203173f7e5119c27 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Fri, 20 Jan 2023 20:13:43 -0500 Subject: [PATCH] correctly import diffusers models by their local path - Corrects a bug in which the local path was treated as a repo_id --- ldm/invoke/CLI.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ldm/invoke/CLI.py b/ldm/invoke/CLI.py index 4891004d66..0de25bf458 100644 --- a/ldm/invoke/CLI.py +++ b/ldm/invoke/CLI.py @@ -577,7 +577,7 @@ def import_model(model_path:str, gen, opt, completer): elif re.match('^[\w.+-]+/[\w.+-]+$',model_path): model_name = import_diffuser_model(model_path, gen, opt, completer) elif os.path.isdir(model_path): - model_name = import_diffuser_model(model_path, gen, opt, completer) + model_name = import_diffuser_model(Path(model_path), gen, opt, completer) else: print(f'** {model_path} is neither the path to a .ckpt file nor a diffusers repository id. Can\'t import.')