From 929dbc135ef0b3443e44d70666ec6dc8872d3c27 Mon Sep 17 00:00:00 2001
From: zhizhen man <dubbocommitter@gmail.com>
Date: Fri, 7 Mar 2025 00:06:18 +0800
Subject: [PATCH] Because the windows file name cannot have an *, replace the *
 with an x

---
 generate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/generate.py b/generate.py
index e8e226d..0e81f0c 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'
-            safe_size = args.size.replace("*", "x")
+            safe_size = args.size.replace("*", "x") if os.name == 'nt' else args.size
             args.save_file = f"{args.task}_{safe_size}_{args.ulysses_size}_{args.ring_size}_{formatted_prompt}_{formatted_time}" + suffix
 
         if "t2i" in args.task: