Allow gguf text encoders and vae

This commit is contained in:
cmdr2
2025-07-14 18:03:39 +05:30
parent 78538f5cfb
commit 8da94496ed

View File

@ -27,7 +27,7 @@ KNOWN_MODEL_TYPES = [
] ]
MODEL_EXTENSIONS = { MODEL_EXTENSIONS = {
"stable-diffusion": [".ckpt", ".safetensors", ".sft", ".gguf"], "stable-diffusion": [".ckpt", ".safetensors", ".sft", ".gguf"],
"vae": [".vae.pt", ".ckpt", ".safetensors", ".sft"], "vae": [".vae.pt", ".ckpt", ".safetensors", ".sft", ".gguf"],
"hypernetwork": [".pt", ".safetensors", ".sft"], "hypernetwork": [".pt", ".safetensors", ".sft"],
"gfpgan": [".pth"], "gfpgan": [".pth"],
"realesrgan": [".pth"], "realesrgan": [".pth"],
@ -35,7 +35,7 @@ MODEL_EXTENSIONS = {
"codeformer": [".pth"], "codeformer": [".pth"],
"embeddings": [".pt", ".bin", ".safetensors", ".sft"], "embeddings": [".pt", ".bin", ".safetensors", ".sft"],
"controlnet": [".pth", ".safetensors", ".sft"], "controlnet": [".pth", ".safetensors", ".sft"],
"text-encoder": [".safetensors", ".sft"], "text-encoder": [".safetensors", ".sft", ".gguf"],
} }
DEFAULT_MODELS = { DEFAULT_MODELS = {
"stable-diffusion": [ "stable-diffusion": [