diff --git a/models/turbine_models/custom_models/sd_inference/utils.py b/models/turbine_models/custom_models/sd_inference/utils.py
index d66515632..cef93bba5 100644
--- a/models/turbine_models/custom_models/sd_inference/utils.py
+++ b/models/turbine_models/custom_models/sd_inference/utils.py
@@ -7,7 +7,6 @@
     PNDMScheduler,
 )
 
-
 def save_external_weights(
     mapper,
     model,
diff --git a/models/turbine_models/custom_models/sdxl_inference/vae.py b/models/turbine_models/custom_models/sdxl_inference/vae.py
index e10664091..dada05d3e 100644
--- a/models/turbine_models/custom_models/sdxl_inference/vae.py
+++ b/models/turbine_models/custom_models/sdxl_inference/vae.py
@@ -57,8 +57,8 @@
 parser.add_argument("--variant", type=str, default="decode")
 parser.add_argument(
     "--decomp_attn",
-    type=argparse.BooleanOptionalAction,
     default=False,
+    action="store_true",
     help="Decompose attention at fx graph level",
 )