From f456bfb385673158e5931b9ecee794e8e2af5ea7 Mon Sep 17 00:00:00 2001
From: ChairC <974833488@qq.com>
Date: Fri, 19 Dec 2025 19:54:16 +0800
Subject: [PATCH] fix: Fix check_parse_image_size_type method modification not
 replaced error.

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

diff --git a/iddm/tools/generate.py b/iddm/tools/generate.py
index 0d6cbb8..034d4aa 100644
--- a/iddm/tools/generate.py
+++ b/iddm/tools/generate.py
@@ -245,7 +245,7 @@ def init_generate_args():
     # Input image size (required)
     # [Warn] Compatible with older versions
     # [Warn] Version <= 1.1.1 need to be equal to model's image size, version > 1.1.1 can set whatever you want
-    parser.add_argument("--image_size", "-i", type=parse_image_size_type, default=64)
+    parser.add_argument("--image_size", "-i", type=check_parse_image_size_type, default=64)
     # Set the use GPU in generate (required)
     parser.add_argument("--use_gpu", type=int, default=0)