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}")