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:51:45 -08:00
parent 91c75a2d63
commit 79fa80a346
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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