smoothieAI
/

pipeline_animatediff_context_controlnet

Model card Files Files and versions Community

smoothieAI commited on Jan 25, 2024

Commit

98d87b9

·

verified ·

1 Parent(s): 68a7452

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -2

pipeline.py CHANGED Viewed

@@ -1303,8 +1303,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                             control_model_input,
                             t,
                             encoder_hidden_states=controlnet_prompt_embeds.to(device),
-                            controlnet_cond=current_context_conditioning_frames.to(device),
-                            conditioning_scale=cond_scale.to(device),
                             guess_mode=guess_mode,
                             return_dict=False,
                         )

                             control_model_input,
                             t,
                             encoder_hidden_states=controlnet_prompt_embeds.to(device),
+                            controlnet_cond=current_context_conditioning_frames,
+                            conditioning_scale=cond_scale,
                             guess_mode=guess_mode,
                             return_dict=False,
                         )