From 2b19a132a6f84bcda48b193b3cfd2c2eb2d2785d Mon Sep 17 00:00:00 2001 From: wl-zhao Date: Sat, 26 Oct 2024 11:07:15 +0800 Subject: [PATCH 1/2] addmap_legacy --- dependency_checker.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/dependency_checker.py b/dependency_checker.py index 73a3cbc..5ff38c5 100644 --- a/dependency_checker.py +++ b/dependency_checker.py @@ -136,7 +136,11 @@ def resolve_dependencies(prompt, custom_dependencies): # resolve custom nodes an pattern = item["field_name"] if re.match(f"^{pattern}$", field_name) and any([filename.endswith(possible_suffix) for possible_suffix in model_suffix]): ckpt_path = get_full_path_or_raise(item["save_path"], filename) - rel_save_path = os.path.relpath(folder_paths.folder_names_and_paths[item["save_path"]][0][0], folder_paths.models_dir) + if hasattr(folder_paths, "map_legacy"): + save_folder = folder_paths.map_legacy(item["save_path"]) + else: + save_folder = item["save_path"] + rel_save_path = os.path.relpath(folder_paths.folder_names_and_paths[save_folder][0][0], folder_paths.models_dir) ckpt_paths[ckpt_path] = { "filename": filename, "rel_save_path": rel_save_path From 752a0de95d5325b46430a0f03856eaadc5bf76bf Mon Sep 17 00:00:00 2001 From: wl-zhao Date: Sun, 27 Oct 2024 15:39:16 +0800 Subject: [PATCH 2/2] add gguf --- dependency_checker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dependency_checker.py b/dependency_checker.py index 5ff38c5..f20a7b7 100644 --- a/dependency_checker.py +++ b/dependency_checker.py @@ -18,7 +18,7 @@ model_loaders_info = json.load(open(os.path.join(os.path.dirname(__file__), "mod node_deps_info = json.load(open(os.path.join(os.path.dirname(__file__), "node_deps_info.json"))) node_blacklist = json.load(open(os.path.join(os.path.dirname(__file__), "node_blacklist.json"))) -model_suffix = [".ckpt", ".safetensors", ".bin", ".pth", ".pt", ".onnx"] +model_suffix = [".ckpt", ".safetensors", ".bin", ".pth", ".pt", ".onnx", ".gguf"] def get_full_path_or_raise(folder_name: str, filename: str) -> str: