From 940cbd749a63e794e1c4b384c528cc44d0e07e60 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Wed, 27 Mar 2024 16:15:09 +0000 Subject: [PATCH] fix arparse --- src/brevitas_examples/stable_diffusion/main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/brevitas_examples/stable_diffusion/main.py b/src/brevitas_examples/stable_diffusion/main.py index 538f1a0d7..934c79edc 100644 --- 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,