Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-06-07 07:44:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

--batch_size parameter for multiple videos without reloading the model.

Browse Source
This commit is contained in:
pftq 2025-02-26 07:04:46 -08:00 committed by GitHub
parent b15339baa3
commit b5d66656ae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
generate.py
View File

@ -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}")