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",