From 4d548bfb5a43ef86545264e60177a75ab9191835 Mon Sep 17 00:00:00 2001
From: Xumin Yu <1090414006@qq.com>
Date: Wed, 6 Nov 2024 20:39:39 +0800
Subject: [PATCH] support .sft for model suffix

---
 dependency_checker.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dependency_checker.py b/dependency_checker.py
index f02cdce..02f53e6 100644
--- a/dependency_checker.py
+++ b/dependency_checker.py
@@ -20,7 +20,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", ".gguf"]
+model_suffix = [".ckpt", ".safetensors", ".bin", ".pth", ".pt", ".onnx", ".gguf", ".sft"]
 extra_packages = ["transformers", "timm", "diffusers", "accelerate"]