From 9e47cce40f3c5c9340e6062786b91a63f602a17f Mon Sep 17 00:00:00 2001 From: zhizhen man Date: Thu, 6 Mar 2025 22:57:46 +0800 Subject: [PATCH] Because the windows file name cannot have an *, replace the * with an x --- generate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/generate.py b/generate.py index f27bb98..e8e226d 100644 --- a/generate.py +++ b/generate.py @@ -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}")