From 20b24f418d30cc7f4062e712fdaa658a33a5853a Mon Sep 17 00:00:00 2001 From: Zhe Kong <43921423+kongzhecn@users.noreply.github.com> Date: Thu, 21 Aug 2025 10:12:44 +0800 Subject: [PATCH] Add files via upload --- generate_infinitetalk.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/generate_infinitetalk.py b/generate_infinitetalk.py index e6383d0..e83daa8 100644 --- a/generate_infinitetalk.py +++ b/generate_infinitetalk.py @@ -82,6 +82,12 @@ def _parse_args(): default=81, help="How many frames to be generated in one clip. The number should be 4n+1" ) + parser.add_argument( + "--max_frame_num", + type=int, + default=1000, + help="The max frame lenght of the generated video." + ) parser.add_argument( "--ckpt_dir", type=str, @@ -630,7 +636,7 @@ def generate(args): audio_guide_scale=args.sample_audio_guide_scale, seed=args.base_seed, offload_model=args.offload_model, - max_frames_num=args.frame_num if args.mode == 'clip' else 1000, + max_frames_num=args.frame_num if args.mode == 'clip' else args.max_frame_num, color_correction_strength = args.color_correction_strength, extra_args=args, )