From 79fa80a346b99d673195a0cd825c83dba22d0014 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 22:51:45 -0800 Subject: [PATCH] Update forge_svd.py --- extensions-builtin/sd_forge_svd/scripts/forge_svd.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py index da97e9a4..bdb3a881 100644 --- a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py +++ b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py @@ -8,11 +8,14 @@ from modules.ui_common import ToolButton, refresh_symbol from modules import shared from modules_forge.gradio_compile import gradio_compile -from ldm_patched.contrib.external import KSampler +from ldm_patched.contrib.external_video_model import ImageOnlyCheckpointLoader, VideoLinearCFGGuidance, SVD_img2vid_Conditioning +from ldm_patched.contrib.external import KSampler, VAEDecode +gradio_compile(SVD_img2vid_Conditioning.INPUT_TYPES(), prefix='condition') gradio_compile(KSampler.INPUT_TYPES(), prefix='sample') - +gradio_compile(VideoLinearCFGGuidance.INPUT_TYPES(), prefix='guidance') +a = 0 svd_root = os.path.join(models_path, 'svd') os.makedirs(svd_root, exist_ok=True)