Update pipeline.py
Browse files- 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,
|