lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-20 14:53:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update forge_svd.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-26 00:03:13 -08:00
parent f4edf7c19a
commit 9ed196d706
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,3 +1,4 @@
import torch
import gradio as gr
import os
import pathlib
@@ -46,8 +47,10 @@ def predict(filename, width, height, video_frames, motion_bucket_id, fps, augmen
model = opVideoLinearCFGGuidance.patch(model_raw, guidance_min_cfg)[0]
init_image = numpy_to_pytorch(input_image)
positive, negative, latent_image = opSVD_img2vid_Conditioning.encode(clip_vision, init_image, vae, width, height, video_frames, motion_bucket_id, fps, augmentation_level)
del model_raw, _, vae, clip_vision, model, init_image, positive, negative, latent_image
model_management.unload_all_models()
model_management.soft_empty_cache()
torch.cuda.empty_cache()
output_latent = opKSampler.sample(model, sampling_seed, sampling_steps, sampling_cfg, sampling_sampler_name, sampling_scheduler, positive, negative, latent_image, sampling_denoise)
output_pixels = opVAEDecode.decode(vae, output_latent)[0]
return