From b5d66656ae2125b1d9d65902d3830b440b9d6e9d Mon Sep 17 00:00:00 2001 From: pftq Date: Wed, 26 Feb 2025 07:04:46 -0800 Subject: [PATCH] --batch_size parameter for multiple videos without reloading the model. --- generate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generate.py b/generate.py index 8bfd79e..a4e8d95 100644 --- a/generate.py +++ b/generate.py @@ -405,7 +405,7 @@ def generate(args): # 20250226 pftq: if multiple videos per batch, change the filename index if args.batch_size > 1: if batch_index == 0: args.save_file = args.save_file.replace(".mp4", "__"+str(batch_index+1)+".mp4") - else: args.save_file = args.save_file.replace("__"+str(batch_index)+".mp4", str(batch_index+1)+".mp4") + else: args.save_file = args.save_file.replace("__"+str(batch_index)+".mp4", "__"+str(batch_index+1)+".mp4") if "t2i" in args.task: logging.info(f"Saving generated image to {args.save_file}")