smoothieAI commited on
Commit
f947467
·
verified ·
1 Parent(s): e764030

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -176,7 +176,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
176
  """
177
 
178
  model_cpu_offload_seq = "text_encoder->image_encoder->unet->vae"
179
- _optional_components = ["feature_extractor", "image_encoder"]
180
 
181
  def __init__(
182
  self,
 
176
  """
177
 
178
  model_cpu_offload_seq = "text_encoder->image_encoder->unet->vae"
179
+ _optional_components = ["feature_extractor", "image_encoder","controlnet"]
180
 
181
  def __init__(
182
  self,