From 2ef4e2184eed1749f71db7ed530e95e9b1e98214 Mon Sep 17 00:00:00 2001 From: zyckk4 Date: Mon, 3 Mar 2025 13:48:19 +0800 Subject: [PATCH] Fix an error saving videos on Windows (#128) --- generate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generate.py b/generate.py index f27bb98..85e516f 100644 --- a/generate.py +++ b/generate.py @@ -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}")