From e945be8e3ebc3b911e8c7a0e527e32bbf2980cc3 Mon Sep 17 00:00:00 2001
From: Christian Bastian <80225746+cdb-boop@users.noreply.github.com>
Date: Mon, 19 Aug 2024 07:29:58 -0400
Subject: [PATCH] Added .gguf to known model extensions.

---
 __init__.py          | 2 +-
 web/model-manager.js | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index 371ecb0..93511fc 100644
--- a/__init__.py
+++ b/__init__.py
@@ -33,7 +33,7 @@ no_preview_image = os.path.join(extension_uri, "no-preview.png")
 ui_settings_uri = os.path.join(extension_uri, "ui_settings.yaml")
 server_settings_uri = os.path.join(extension_uri, "server_settings.yaml")
 
-fallback_model_extensions = set([".bin", ".ckpt", ".onnx", ".pt", ".pth", ".safetensors"]) # TODO: magic values
+fallback_model_extensions = set([".bin", ".ckpt", ".gguf", ".onnx", ".pt", ".pth", ".safetensors"]) # TODO: magic values
 jpeg_format_names = ["JPG", "JPEG", "JFIF"]
 image_extensions = (
     ".png", # order matters
diff --git a/web/model-manager.js b/web/model-manager.js
index 1207eef..a9246a3 100644
--- a/web/model-manager.js
+++ b/web/model-manager.js
@@ -157,7 +157,7 @@ const modelNodeType = {
     "vae_approx": undefined,
 };
 
-const MODEL_EXTENSIONS = [".bin", ".ckpt", ".onnx", ".pt", ".pth", ".safetensors"]; // TODO: ask server for?
+const MODEL_EXTENSIONS = [".bin", ".ckpt", "gguf", ".onnx", ".pt", ".pth", ".safetensors"]; // TODO: ask server for?
 const IMAGE_EXTENSIONS = [
     ".png", 
     ".webp",