Update pipeline.py
Browse files- pipeline.py +4 -1
pipeline.py
CHANGED
@@ -136,13 +136,16 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
136 |
EulerAncestralDiscreteScheduler,
|
137 |
DPMSolverMultistepScheduler,
|
138 |
],
|
139 |
-
controlnet: Union[ControlNetModel, MultiControlNetModel] = None,
|
140 |
feature_extractor: CLIPImageProcessor = None,
|
141 |
image_encoder: CLIPVisionModelWithProjection = None,
|
142 |
):
|
143 |
super().__init__()
|
144 |
unet = UNetMotionModel.from_unet2d(unet, motion_adapter)
|
145 |
|
|
|
|
|
|
|
146 |
self.register_modules(
|
147 |
vae=vae,
|
148 |
text_encoder=text_encoder,
|
|
|
136 |
EulerAncestralDiscreteScheduler,
|
137 |
DPMSolverMultistepScheduler,
|
138 |
],
|
139 |
+
controlnet: Union[ControlNetModel, List[ControlNetModel], Tuple[ControlNetModel], MultiControlNetModel] = None,
|
140 |
feature_extractor: CLIPImageProcessor = None,
|
141 |
image_encoder: CLIPVisionModelWithProjection = None,
|
142 |
):
|
143 |
super().__init__()
|
144 |
unet = UNetMotionModel.from_unet2d(unet, motion_adapter)
|
145 |
|
146 |
+
if isinstance(controlnet, (list, tuple)):
|
147 |
+
controlnet = MultiControlNetModel(controlnet)
|
148 |
+
|
149 |
self.register_modules(
|
150 |
vae=vae,
|
151 |
text_encoder=text_encoder,
|