smoothieAI commited on
Commit
f6f9bc5
·
verified ·
1 Parent(s): 14746ed

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +23 -19
pipeline.py CHANGED
@@ -199,26 +199,30 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
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
- # print all the attributes
207
- print("Attributes:")
208
- for attr in dir(self):
209
- print(attr)
210
-
211
- self.register_modules(
212
- vae=vae,
213
- text_encoder=text_encoder,
214
- tokenizer=tokenizer,
215
- unet=unet,
216
- motion_adapter=motion_adapter,
217
- controlnet=controlnet,
218
- scheduler=scheduler,
219
- feature_extractor=feature_extractor,
220
- image_encoder=image_encoder,
221
- )
 
 
 
 
222
  self.vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1)
223
  self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor)
224
  self.control_image_processor = VaeImageProcessor(
 
199
  ):
200
  super().__init__()
201
  unet = UNetMotionModel.from_unet2d(unet, motion_adapter)
202
+
203
  if controlnet is None:
204
+ self.register_modules(
205
+ vae=vae,
206
+ text_encoder=text_encoder,
207
+ tokenizer=tokenizer,
208
+ unet=unet,
209
+ motion_adapter=motion_adapter,
210
+ scheduler=scheduler,
211
+ feature_extractor=feature_extractor,
212
+ image_encoder=image_encoder,
213
+ )
214
+ else:
215
+ self.register_modules(
216
+ vae=vae,
217
+ text_encoder=text_encoder,
218
+ tokenizer=tokenizer,
219
+ unet=unet,
220
+ motion_adapter=motion_adapter,
221
+ controlnet=controlnet,
222
+ scheduler=scheduler,
223
+ feature_extractor=feature_extractor,
224
+ image_encoder=image_encoder,
225
+ )
226
  self.vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1)
227
  self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor)
228
  self.control_image_processor = VaeImageProcessor(