From f49589ee828bfb7b5539ce6d964103378d7e1f98 Mon Sep 17 00:00:00 2001 From: wl-zhao Date: Fri, 18 Oct 2024 10:05:36 +0800 Subject: [PATCH] add printed logs --- dependency_checker.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/dependency_checker.py b/dependency_checker.py index 0f7d5fd..15d62ee 100644 --- a/dependency_checker.py +++ b/dependency_checker.py @@ -12,7 +12,7 @@ from .file_upload import collect_local_file, process_local_file_path_async model_list_json = json.load(open(os.path.join(os.path.dirname(__file__), "model_info.json"))) model_loaders_info = json.load(open(os.path.join(os.path.dirname(__file__), "model_loader_info.json"))) -model_suffix = [".ckpt", ".safetensors", ".bin"] +model_suffix = [".ckpt", ".safetensors", ".bin", ".pth"] def handle_model_info(ckpt_path): ckpt_path = windows_to_linux_path(ckpt_path) @@ -123,6 +123,7 @@ def resolve_dependencies(prompt, custom_dependencies): # resolve custom nodes an if filename.endswith(possible_suffix): is_model = True if is_model: + print(f"find {filename}, is_model=True") # find possible paths matching_files = [] # Walk through all subdirectories and files in the directory @@ -131,6 +132,7 @@ def resolve_dependencies(prompt, custom_dependencies): # resolve custom nodes an if file.endswith(filename): # Add full file path to the result matching_files.append(os.path.join(root, file)) + print(f"matched files: {matching_files}") if len(matching_files) == 1: ckpt_paths.append(matching_files[0]) list(map(partial(collect_local_file, mapping_dict=file_mapping_dict), node_info["inputs"].values()))