diff --git a/infer.py b/infer.py
index 7767e5e..0813fcd 100644
--- a/infer.py
+++ b/infer.py
@@ -165,7 +165,7 @@ def generate_video(
     parser.add_argument("--num_videos_per_prompt", type=int, default=1, help="Number of videos to generate per prompt")
     parser.add_argument("--dtype", type=str, default="bfloat16", help="The data type for computation (e.g., 'float16' or 'bfloat16')")
     parser.add_argument("--seed", type=int, default=42, help="The seed for reproducibility")
-    # auxiliary Model
+    # auxiliary model arguments
     parser.add_argument("--is_upscale", action='store_true', help="Enable video upscaling (super-resolution) if this flag is set.")
     parser.add_argument("--is_frame_interpolation", action='store_true', help="Enable frame interpolation to increase frame rate if this flag is set.")
 
diff --git a/models/pipeline_consisid.py b/models/pipeline_consisid.py
index 1589586..0ac1c01 100644
--- a/models/pipeline_consisid.py
+++ b/models/pipeline_consisid.py
@@ -14,7 +14,7 @@
 
 import inspect
 import math
-from typing import Callable, Dict, List, Optional, Tuple, Union
+from typing import Callable, Any, Dict, List, Optional, Tuple, Union
 
 import cv2
 import numpy as np