From c495d81ba82946c519f612d9f820f4700f1883fd Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Fri, 26 Jan 2024 00:31:07 -0800 Subject: [PATCH] Update forge_svd.py --- extensions-builtin/sd_forge_svd/scripts/forge_svd.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py index 5f9a6943..43615503 100644 --- a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py +++ b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py @@ -51,7 +51,8 @@ def predict(filename, width, height, video_frames, motion_bucket_id, fps, augmen positive, negative, latent_image = opSVD_img2vid_Conditioning.encode(clip_vision, init_image, vae, width, height, video_frames, motion_bucket_id, fps, augmentation_level) output_latent = opKSampler.sample(model, sampling_seed, sampling_steps, sampling_cfg, sampling_sampler_name, sampling_scheduler, positive, negative, latent_image, sampling_denoise)[0] output_pixels = opVAEDecode.decode(vae, output_latent)[0] - return + outputs = pytorch_to_numpy(output_pixels) + return outputs def on_ui_tabs():