JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-06-29 05:39:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

make the style of argparser consistent with repo

Browse Source
This commit is contained in:
LittleNyima 2025-02-23 19:41:21 +08:00
parent 250a0bce45
commit e0bf395458
No known key found for this signature in database
GPG Key ID: 87E0B0CE8AF393AB
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
inference/ddim_inversion.py
View File

@ -50,16 +50,16 @@ class DDIMInversionArguments(TypedDict):
def get_args() -> DDIMInversionArguments:
parser = argparse.ArgumentParser()
parser.add_argument("--model-path", type=str, required=True, help="Path of the pretrained model")
parser.add_argument("--model_path", type=str, required=True, help="Path of the pretrained model")
parser.add_argument("--prompt", type=str, required=True, help="Prompt for the direct sample procedure")
parser.add_argument("--video-path", type=str, required=True, help="Path of the video for inversion")
parser.add_argument("--output-path", type=str, default="output", help="Path of the output videos")
parser.add_argument("--guidance-scale", type=float, default=6.0, help="Classifier-free guidance scale")
parser.add_argument("--num-inference-steps", type=int, default=50, help="Number of inference steps")
parser.add_argument("--skip-frames-start", type=int, default=0, help="Number of skipped frames from the start")
parser.add_argument("--skip-frames-end", type=int, default=0, help="Number of skipped frames from the end")
parser.add_argument("--frame-sample-step", type=int, default=None, help="Temporal stride of the sampled frames")
parser.add_argument("--max-num-frames", type=int, default=81, help="Max number of sampled frames")
parser.add_argument("--video_path", type=str, required=True, help="Path of the video for inversion")
parser.add_argument("--output_path", type=str, default="output", help="Path of the output videos")
parser.add_argument("--guidance_scale", type=float, default=6.0, help="Classifier-free guidance scale")
parser.add_argument("--num_inference_steps", type=int, default=50, help="Number of inference steps")
parser.add_argument("--skip_frames_start", type=int, default=0, help="Number of skipped frames from the start")
parser.add_argument("--skip_frames_end", type=int, default=0, help="Number of skipped frames from the end")
parser.add_argument("--frame_sample_step", type=int, default=None, help="Temporal stride of the sampled frames")
parser.add_argument("--max_num_frames", type=int, default=81, help="Max number of sampled frames")
parser.add_argument("--width", type=int, default=720, help="Resized width of the video frames")
parser.add_argument("--height", type=int, default=480, help="Resized height of the video frames")
parser.add_argument("--fps", type=int, default=8, help="Frame rate of the output videos")