From 3c5d8ee73ba344ef42272dac7265094a84ad5d46 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 21:54:40 -0800 Subject: [PATCH] Update forge_svd.py --- extensions-builtin/sd_forge_svd/scripts/forge_svd.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py index 50c91ec1..539a5387 100644 --- a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py +++ b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py @@ -3,10 +3,18 @@ import os from modules import scripts, script_callbacks from modules.paths import models_path +from modules import shared svd_root = os.path.join(models_path, 'svd') os.makedirs(svd_root, exist_ok=True) +svd_filenames = [] + + +def update_svd_filenames(): + global svd_filenames + svd_filenames = list(shared.walk_files(svd_root, allowed_extensions=[".pt", ".ckpt", ".safetensors"])) + return svd_filenames class ForgeSVD(scripts.Script):