Update forge_svd.py

This commit is contained in:
lllyasviel
2024-01-25 23:34:52 -08:00
parent 1919a835a4
commit 7147972e80

View File

@@ -40,7 +40,8 @@ def predict(filename, width, height, video_frames, motion_bucket_id, fps, augmen
sampling_seed, sampling_steps, sampling_cfg, sampling_sampler_name, sampling_scheduler,
sampling_denoise, guidance_min_cfg, input_image):
filename = os.path.join(svd_root, filename)
model, _, vae, clip_vision = load_checkpoint_guess_config(filename, output_vae=True, output_clip=False, output_clipvision=True)
model_raw, _, vae, clip_vision = load_checkpoint_guess_config(filename, output_vae=True, output_clip=False, output_clipvision=True)
model = opVideoLinearCFGGuidance.patch(model_raw, guidance_min_cfg)
a = 0
return