From 86c56bb919915752b45c38dbdfd1ef99f47a2525 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 21:51:57 -0800 Subject: [PATCH] Update forge_svd.py --- extensions-builtin/sd_forge_svd/scripts/forge_svd.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py index f494378c..50c91ec1 100644 --- a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py +++ b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py @@ -1,6 +1,12 @@ import gradio as gr +import os from modules import scripts, script_callbacks +from modules.paths import models_path + + +svd_root = os.path.join(models_path, 'svd') +os.makedirs(svd_root, exist_ok=True) class ForgeSVD(scripts.Script):