smoothieAI commited on
Commit
cd11b61
·
verified ·
1 Parent(s): 80fffd8

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -0
pipeline.py CHANGED
@@ -199,6 +199,9 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
199
  ):
200
  super().__init__()
201
  unet = UNetMotionModel.from_unet2d(unet, motion_adapter)
 
 
 
202
 
203
  self.register_modules(
204
  vae=vae,
 
199
  ):
200
  super().__init__()
201
  unet = UNetMotionModel.from_unet2d(unet, motion_adapter)
202
+
203
+ if controlnet is None:
204
+ if hasattr(self, "controlnet"):delattr(self, "controlnet")
205
 
206
  self.register_modules(
207
  vae=vae,