smoothieAI
/

pipeline_animatediff_context_controlnet

Model card Files Files and versions Community

smoothieAI commited on Jan 25, 2024

Commit

30fe9a2

·

verified ·

1 Parent(s): 98d87b9

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -1300,7 +1300,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                         )
                         down_block_res_samples, mid_block_res_sample = self.controlnet(
-                            control_model_input,
                             t,
                             encoder_hidden_states=controlnet_prompt_embeds.to(device),
                             controlnet_cond=current_context_conditioning_frames,

                         )
                         down_block_res_samples, mid_block_res_sample = self.controlnet(
+                            control_model_input.to(device),
                             t,
                             encoder_hidden_states=controlnet_prompt_embeds.to(device),
                             controlnet_cond=current_context_conditioning_frames,