From 9ffaf82bdfeb02b5367ef1812b45629b32a101f0 Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Mon, 26 Feb 2024 15:49:02 -0600 Subject: [PATCH] Fix --decomp_attn for VAE as well. --- models/turbine_models/custom_models/sd_inference/utils.py | 1 - models/turbine_models/custom_models/sdxl_inference/vae.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) 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", )