smoothieAI commited on
Commit
582c0b8
·
verified ·
1 Parent(s): ba73a29

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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,