From 8da94496ed5ab4e749a3da17206374236e6cca0d Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Mon, 14 Jul 2025 18:03:39 +0530
Subject: [PATCH] Allow gguf text encoders and vae

---
 ui/easydiffusion/model_manager.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ui/easydiffusion/model_manager.py b/ui/easydiffusion/model_manager.py
index d5f7fde3..2b323162 100644
--- a/ui/easydiffusion/model_manager.py
+++ b/ui/easydiffusion/model_manager.py
@@ -27,7 +27,7 @@ KNOWN_MODEL_TYPES = [
 ]
 MODEL_EXTENSIONS = {
     "stable-diffusion": [".ckpt", ".safetensors", ".sft", ".gguf"],
-    "vae": [".vae.pt", ".ckpt", ".safetensors", ".sft"],
+    "vae": [".vae.pt", ".ckpt", ".safetensors", ".sft", ".gguf"],
     "hypernetwork": [".pt", ".safetensors", ".sft"],
     "gfpgan": [".pth"],
     "realesrgan": [".pth"],
@@ -35,7 +35,7 @@ MODEL_EXTENSIONS = {
     "codeformer": [".pth"],
     "embeddings": [".pt", ".bin", ".safetensors", ".sft"],
     "controlnet": [".pth", ".safetensors", ".sft"],
-    "text-encoder": [".safetensors", ".sft"],
+    "text-encoder": [".safetensors", ".sft", ".gguf"],
 }
 DEFAULT_MODELS = {
     "stable-diffusion": [