fix arparse

Xilinx · Mar 27, 2024 · 940cbd7 · 940cbd7
1 parent 6c74db3
commit 940cbd7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/brevitas_examples/stable_diffusion/main.py b/src/brevitas_examples/stable_diffusion/main.py
@@ -289,7 +289,7 @@ def input_bit_width(module):
     add_bool_arg(parser, 'float16', default=True, help='Enable float16 execution.')
     add_bool_arg(parser, 'attention-slicing', default=False, help='Enable attention slicing.')
     add_bool_arg(
-        parser, 'is_sd_xl', default=False, help='Enable this flag to correctly export SDXL')
+        parser, 'is-sd-xl', default=False, help='Enable this flag to correctly export SDXL')
     parser.add_argument(
         '--export-target',
         type=str,
@@ -306,9 +306,9 @@ def input_bit_width(module):
     parser.add_argument(
         '--linear-weight-bit-width', type=int, default=8, help='Weight bit width. Default: 8.')
     parser.add_argument(
-        '--conv-input-bit-width', type=int, default=None, help='Input bit width. Default: 8.')
+        '--conv-input-bit-width', type=int, default=8, help='Input bit width. Default: 8.')
     parser.add_argument(
-        '--linear-input-bit-width', type=int, default=None, help='Input bit width. Default: 8.')
+        '--linear-input-bit-width', type=int, default=8, help='Input bit width. Default: 8.')
     parser.add_argument(
         '--weight-param-method',
         type=str,