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(
|