lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-05 13:09:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update forge_svd.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 21:54:40 -08:00
parent 86c56bb919
commit 3c5d8ee73b
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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