soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages 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
Show Stats 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 gradio as gr
import os import os
import pathlib
from modules import scripts, script_callbacks from modules import scripts, script_callbacks
from modules.paths import models_path from modules.paths import models_path
@ -13,7 +14,7 @@ svd_filenames = []
def update_svd_filenames(): def update_svd_filenames():
global 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 return svd_filenames