smoothieAI commited on
Commit
e0a707a
·
verified ·
1 Parent(s): 0a7ee5b

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +4 -2
pipeline.py CHANGED
@@ -48,7 +48,10 @@ from diffusers.models import ControlNetModel
48
  from diffusers.pipelines.controlnet.multicontrolnet import MultiControlNetModel
49
  from diffusers.pipelines.pipeline_utils import DiffusionPipeline
50
  from diffusers.utils import deprecate
51
-
 
 
 
52
 
53
  logger = logging.get_logger(__name__) # pylint: disable=invalid-name
54
 
@@ -962,7 +965,6 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
962
  prompt_embeds.dtype,
963
  device,
964
  generator,
965
- latents=None,
966
  )
967
 
968
  # 6. Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline
 
48
  from diffusers.pipelines.controlnet.multicontrolnet import MultiControlNetModel
49
  from diffusers.pipelines.pipeline_utils import DiffusionPipeline
50
  from diffusers.utils import deprecate
51
+ import PIL
52
+ from PIL import Image
53
+ import torchvision
54
+ import math
55
 
56
  logger = logging.get_logger(__name__) # pylint: disable=invalid-name
57
 
 
965
  prompt_embeds.dtype,
966
  device,
967
  generator,
 
968
  )
969
 
970
  # 6. Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline