smoothieAI
/

pipeline_animatediff_context_controlnet

Model card Files Files and versions Community

smoothieAI commited on Jan 25, 2024

Commit

ba73a29

·

verified ·

1 Parent(s): 779041e

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -3

pipeline.py CHANGED Viewed

@@ -136,7 +136,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
             EulerAncestralDiscreteScheduler,
             DPMSolverMultistepScheduler,
         ],
-        controlnet: Optional[Union[ControlNetModel, MultiControlNetModel]] = None,
         feature_extractor: CLIPImageProcessor = None,
         image_encoder: CLIPVisionModelWithProjection = None,
     ):
@@ -1384,5 +1384,4 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
         if not return_dict:
             return (video,)
-        return AnimateDiffPipelineOutput(frames=video)

             EulerAncestralDiscreteScheduler,
             DPMSolverMultistepScheduler,
         ],
+        controlnet: Union[ControlNetModel, MultiControlNetModel] = None,
         feature_extractor: CLIPImageProcessor = None,
         image_encoder: CLIPVisionModelWithProjection = None,
     ):
         if not return_dict:
             return (video,)
+        return AnimateDiffPipelineOutput(frames=video)