diff --git a/animatediff/pipelines/pipeline_animation.py b/animatediff/pipelines/pipeline_animation.py
index 1193c71..7cc9226 100644
--- a/animatediff/pipelines/pipeline_animation.py
+++ b/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)
\ No newline at end of file
+        return AnimationPipelineOutput(videos=video)
diff --git a/animatediff/utils/convert_from_ckpt.py b/animatediff/utils/convert_from_ckpt.py
index 9c70b92..9ee269d 100644
--- a/animatediff/utils/convert_from_ckpt.py
+++ b/animatediff/utils/convert_from_ckpt.py
@@ -33,7 +33,6 @@ from transformers import (
 
 from diffusers.models import (
     AutoencoderKL,
-    # ControlNetModel,
     PriorTransformer,
     UNet2DConditionModel,
 )