From 41be4956729b52b4024a8cceed8c2334a7434807 Mon Sep 17 00:00:00 2001 From: lilly Date: Thu, 6 Apr 2023 21:29:47 +0900 Subject: [PATCH] Update CheckpointLoaderSimpleText.py --- CheckpointLoaderSimpleText.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/CheckpointLoaderSimpleText.py b/CheckpointLoaderSimpleText.py index 942919c..4fce737 100644 --- a/CheckpointLoaderSimpleText.py +++ b/CheckpointLoaderSimpleText.py @@ -43,7 +43,10 @@ class CheckpointLoaderSimpleText: print(f"ckpt_path",ckpt_path) if ckpt_path is None: print(f"{ckpt_name} is none") - (name,fullpath)=filenameget(os.path.join(models_dir, "checkpoints")+"/**/"+ckpt_name+"*.safetensors") + if ckpt_name.endswith(".safetensors") or ckpt_name.endswith(".ckpt"): + (name,fullpath)=filenameget(os.path.join(models_dir, "checkpoints")+"/**/"+ckpt_name) + else: + (name,fullpath)=filenameget(os.path.join(models_dir, "checkpoints")+"/**/"+ckpt_name+"*.safetensors") if fullpath is None: print(f"{name} is none") return