Update forge_svd.py
This commit is contained in:
parent
156188a17c
commit
a0694c856b
@ -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
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user