Fix an error saving videos on Windows (#128)

This commit is contained in:
zyckk4 2025-03-03 13:48:19 +08:00 committed by GitHub
parent a326079926
commit 2ef4e2184e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -384,7 +384,7 @@ 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
args.save_file = f"{args.task}_{args.size.replace('*','x') if sys.platform=='win32' else args.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}")