support .sft for model suffix
This commit is contained in:
@@ -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"]
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user