From 104ccda2764a2d9c10f81642c735c254b2dfa5af Mon Sep 17 00:00:00 2001
From: butaixianran <butaixianran@qq.com>
Date: Sat, 11 Mar 2023 12:38:18 +0800
Subject: [PATCH] ignore .vae file in model folder

---
 scripts/lib/model.py                | 2 +-
 scripts/lib/model_action_civitai.py | 7 +++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/scripts/lib/model.py b/scripts/lib/model.py
index a264415..efdb1fd 100644
--- a/scripts/lib/model.py
+++ b/scripts/lib/model.py
@@ -21,7 +21,7 @@ folders = {
 
 exts = (".bin", ".pt", ".safetensors", ".ckpt")
 info_ext = ".info"
-
+vae_suffix = ".vae"
 
 
 # get cusomter model path
diff --git a/scripts/lib/model_action_civitai.py b/scripts/lib/model_action_civitai.py
index fe0a014..d188491 100644
--- a/scripts/lib/model_action_civitai.py
+++ b/scripts/lib/model_action_civitai.py
@@ -24,6 +24,13 @@ def scan_model(max_size_preview, skip_nsfw_preview):
                 item = os.path.join(root, filename)
                 base, ext = os.path.splitext(item)
                 if ext in model.exts:
+                    # ignore vae file
+                    if len(base) > 4:
+                        if base[-4:] == model.vae_suffix:
+                            # find .vae
+                            util.printD("This is a vae file: " + filename)
+                            continue
+
                     # find a model
                     # get info file
                     info_file = base + civitai.suffix + model.info_ext