From 178773a43a59c8dacad36560150e0f2f38c01e4a Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 5 Jan 2024 12:06:17 +0300 Subject: [PATCH] Cleanup --- modules/processing_vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/processing_vae.py b/modules/processing_vae.py index d07c7ae06..58b6b0ab1 100644 --- a/modules/processing_vae.py +++ b/modules/processing_vae.py @@ -49,7 +49,7 @@ def full_vae_decode(latents, model): latents = latents.to(next(iter(model.vae.post_quant_conv.parameters())).dtype) # OpenVINO with INT4 doesn't work with VAE decode so we pass that we are using VAE right now to OpenVINO - if shared.compiled_model_state is not None: + if shared.opts.cuda_compile and shared.opts.cuda_compile_backend == "openvino_fx" and shared.compiled_model_state.first_pass_vae: shared.compiled_model_state.compiling_vae = True decoded = model.vae.decode(latents / model.vae.config.scaling_factor, return_dict=False)[0]