Update pipeline.py
Browse files- pipeline.py +1 -4
pipeline.py
CHANGED
@@ -193,15 +193,12 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
193 |
EulerAncestralDiscreteScheduler,
|
194 |
DPMSolverMultistepScheduler,
|
195 |
],
|
196 |
-
|
197 |
feature_extractor: Optional[CLIPImageProcessor] = None,
|
198 |
image_encoder: Optional[CLIPVisionModelWithProjection] = None,
|
199 |
):
|
200 |
super().__init__()
|
201 |
unet = UNetMotionModel.from_unet2d(unet, motion_adapter)
|
202 |
-
|
203 |
-
# temp workaround to prevent ip adapter library from loading ip adapter on empty controlnet parameter
|
204 |
-
controlnet = controlnets
|
205 |
|
206 |
if controlnet is None:
|
207 |
self.register_modules(
|
|
|
193 |
EulerAncestralDiscreteScheduler,
|
194 |
DPMSolverMultistepScheduler,
|
195 |
],
|
196 |
+
controlnet: Optional[Union[ControlNetModel, MultiControlNetModel]]=None,
|
197 |
feature_extractor: Optional[CLIPImageProcessor] = None,
|
198 |
image_encoder: Optional[CLIPVisionModelWithProjection] = None,
|
199 |
):
|
200 |
super().__init__()
|
201 |
unet = UNetMotionModel.from_unet2d(unet, motion_adapter)
|
|
|
|
|
|
|
202 |
|
203 |
if controlnet is None:
|
204 |
self.register_modules(
|