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 23:04:14 -08:00
parent cf1f31351e
commit a3ec20b03f
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -29,6 +29,12 @@ def update_svd_filenames():
return svd_filenames
def predict(filename, width, height, video_frames, motion_bucket_id, fps, augmentation_level,
sampling_seed, sampling_steps, sampling_cfg, sampling_sampler_name, sampling_scheduler,
sampling_denoise, guidance_min_cfg):
return
class ForgeSVD(scripts.Script):
def __init__(self) -> None:
super().__init__()
@ -79,16 +85,18 @@ def on_ui_tabs():
'ddim_uniform'], value='normal')
sampling_denoise = gr.Slider(label='Sampling Denoise', minimum=0.0, maximum=1.0, step=0.01, value=1.0)
guidance_min_cfg = gr.Slider(label='Guidance Min Cfg', minimum=0.0, maximum=100.0, step=0.5, value=1.0)
generate_button = gr.Button(value="Generate")
ctrls = [filename, width, height, video_frames, motion_bucket_id, fps, augmentation_level,
sampling_seed, sampling_steps, sampling_cfg, sampling_sampler_name, sampling_scheduler,
sampling_denoise, guidance_min_cfg, generate_button]
sampling_denoise, guidance_min_cfg]
with gr.Column():
output_gallery = gr.Gallery(label='Gallery', show_label=False, object_fit='contain',
visible=True, height=1024, columns=4)
generate_button.click(predict, inputs=ctrls, outputs=[output_gallery])
return [(svd_block, "SVD", "svd")]