lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-02 22:37:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update forge_svd.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 22:00:36 -08:00
parent 156188a17c
commit a0694c856b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
extensions-builtin/sd_forge_svd/scripts/forge_svd.py
View File

@@ -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