smoothieAI commited on
Commit
e4e1338
1 Parent(s): 941962f

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -4
pipeline.py CHANGED
@@ -193,15 +193,12 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
193
  EulerAncestralDiscreteScheduler,
194
  DPMSolverMultistepScheduler,
195
  ],
196
- controlnets: 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
- # 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(