From 32fc090a43d24939c33d2de3eee42c2141b9f5a0 Mon Sep 17 00:00:00 2001 From: saienduri <77521230+saienduri@users.noreply.github.com> Date: Mon, 22 Jul 2024 14:11:12 -0700 Subject: [PATCH] change preprocess vae check --- models/turbine_models/custom_models/sd_inference/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/turbine_models/custom_models/sd_inference/utils.py b/models/turbine_models/custom_models/sd_inference/utils.py index 24af34a6..1caba490 100644 --- a/models/turbine_models/custom_models/sd_inference/utils.py +++ b/models/turbine_models/custom_models/sd_inference/utils.py @@ -252,7 +252,7 @@ def compile_to_vmfb( flags.extend(MI_flags["pad_attention"]) elif "punet" in flagset_keywords: flags.extend(MI_flags["punet"]) - elif "vae_preprocess" in flagset_keywords: + elif "vae" in safe_name: flags.extend(MI_flags["vae_preprocess"]) else: flags.extend(MI_flags["preprocess_default"])