From f094cae7e9b8d7119e2a560f62a7dc691b5b2f2f Mon Sep 17 00:00:00 2001 From: Artiprocher Date: Fri, 21 Jun 2024 16:24:39 +0800 Subject: [PATCH] fix compatibility issues in sd_video_pipeline --- diffsynth/pipelines/stable_diffusion_video.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/diffsynth/pipelines/stable_diffusion_video.py b/diffsynth/pipelines/stable_diffusion_video.py index eed4da3..b204cad 100644 --- a/diffsynth/pipelines/stable_diffusion_video.py +++ b/diffsynth/pipelines/stable_diffusion_video.py @@ -40,7 +40,7 @@ def lets_dance_with_long_video( sample[batch_id: batch_id_].to(device), timestep, encoder_hidden_states[batch_id: batch_id_].to(device), - controlnet_frames[:, batch_id: batch_id_].to(device) if controlnet_frames is not None else None, + controlnet_frames=controlnet_frames[:, batch_id: batch_id_].to(device) if controlnet_frames is not None else None, unet_batch_size=unet_batch_size, controlnet_batch_size=controlnet_batch_size, cross_frame_attention=cross_frame_attention, device=device, vram_limit_level=vram_limit_level