From 3a940b0bc84f02953a19c416ec36f367f32005c7 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Sun, 2 Feb 2025 03:40:24 -0500 Subject: [PATCH] fps parameter --- animation/StableAnimator/inference_advanced.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/animation/StableAnimator/inference_advanced.py b/animation/StableAnimator/inference_advanced.py index 128a9b7..9855495 100644 --- a/animation/StableAnimator/inference_advanced.py +++ b/animation/StableAnimator/inference_advanced.py @@ -273,6 +273,14 @@ def parse_args(): required=False ) + # Input and output video FPS (these may require separation later) + parser.add_argument( + "--fps", + type=int, + default=30, + required=False + ) + args = parser.parse_args() return args @@ -520,7 +528,7 @@ if __name__ == "__main__": os.makedirs(args.frame_output_dir, exist_ok=True) save_frames_as_png(video_frames, args.frame_output_dir) - merge_frames_to_mp4(args.frame_output_dir, args.video_output_path) + merge_frames_to_mp4(args.frame_output_dir, args.video_output_path, framerate=args.fps) # bash command_basic_infer.sh