From a0694c856bec488b8c8d91d7703ba115fc624f5e Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 22:00:36 -0800 Subject: [PATCH] Update forge_svd.py --- extensions-builtin/sd_forge_svd/scripts/forge_svd.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py index bd2c6e31..bc4c551d 100644 --- a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py +++ b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py @@ -1,5 +1,6 @@ import gradio as gr import os +import pathlib from modules import scripts, script_callbacks from modules.paths import models_path @@ -13,7 +14,7 @@ svd_filenames = [] def update_svd_filenames(): global svd_filenames - svd_filenames = list(shared.walk_files(svd_root, allowed_extensions=[".pt", ".ckpt", ".safetensors"])) + svd_filenames = [pathlib.Path(x).name for x in shared.walk_files(svd_root, allowed_extensions=[".pt", ".ckpt", ".safetensors"])] return svd_filenames