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