support .sft for model suffix

This commit is contained in:
Xumin Yu
2024-11-06 20:39:39 +08:00
committed by GitHub
parent 24c17beccf
commit 4d548bfb5a
+1 -1
View File
@@ -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"]