From e06fb3d5d37bfc15a249e9e3fb969ac128431cf3 Mon Sep 17 00:00:00 2001 From: wl-zhao Date: Fri, 18 Oct 2024 10:25:54 +0800 Subject: [PATCH] update model suffix --- dependency_checker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dependency_checker.py b/dependency_checker.py index c74080b..18328c0 100644 --- a/dependency_checker.py +++ b/dependency_checker.py @@ -13,7 +13,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", ".pth"] +model_suffix = [".ckpt", ".safetensors", ".bin", ".pth", ".pt", ".onnx"] def handle_model_info(ckpt_path): ckpt_path = windows_to_linux_path(ckpt_path)