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