update

animatediff/pipelines/pipeline_animation.py

@@ -1,4 +1,4 @@
-# Adapted from https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
+# Adapted from https://github.com/showlab/Tune-A-Video/blob/main/tuneavideo/pipelines/pipeline_tuneavideo.py
 
 import inspect
 from typing import Callable, List, Optional, Union
@@ -420,4 +420,4 @@ class AnimationPipeline(DiffusionPipeline):
         if not return_dict:
             return video
 
-        return AnimationPipelineOutput(videos=video)
+        return AnimationPipelineOutput(videos=video)

animatediff/utils/convert_from_ckpt.py

@@ -33,7 +33,6 @@ from transformers import (
 
 from diffusers.models import (
     AutoencoderKL,
-    # ControlNetModel,
     PriorTransformer,
     UNet2DConditionModel,
 )