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

Because the windows file name cannot have an *, replace the * with an x

Browse Source
This commit is contained in:
zhizhen man 2025-03-06 22:57:46 +08:00
parent 6797c48002
commit 9e47cce40f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
generate.py
View File

@ -384,7 +384,8 @@ def generate(args):
formatted_prompt = args.prompt.replace(" ", "_").replace("/",
"_")[:50]
suffix = '.png' if "t2i" in args.task else '.mp4'
args.save_file = f"{args.task}_{args.size}_{args.ulysses_size}_{args.ring_size}_{formatted_prompt}_{formatted_time}" + suffix
safe_size = args.size.replace("*", "x")
args.save_file = f"{args.task}_{safe_size}_{args.ulysses_size}_{args.ring_size}_{formatted_prompt}_{formatted_time}" + suffix
if "t2i" in args.task:
logging.info(f"Saving generated image to {args.save_file}")