Fix --decomp_attn for VAE as well.

nod-ai · Feb 26, 2024 · 9ffaf82 · 9ffaf82
1 parent 8b7087c
commit 9ffaf82
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -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",
 )
-Original file line number
+Diff line change
@@ Expand Up / @@ -7,7 +7,6 @@ @@
         PNDMScheduler,
     )
     def save_external_weights(
         mapper,
         model,
@@ Expand Down @@