update

pkuhxy · Dec 22, 2024 · 35a770d · 35a770d
1 parent 3bae6fe
commit 35a770d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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