diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py
index 93420b397..5a52e30df 100644
--- a/modules/sd_hijack.py
+++ b/modules/sd_hijack.py
@@ -189,10 +189,10 @@ class StableDiffusionModelHijack:
                 shared.log.info(f"Compiling pipeline={m.model.__class__.__name__} mode={opts.cuda_compile_backend}")
                 import torch._dynamo # pylint: disable=unused-import,redefined-outer-name
                 if shared.opts.cuda_compile_backend == "openvino_fx":
-                    torch._dynamo.reset()
-                    from modules.intel.openvino import openvino_fx, openvino_clear_caches, model_state # pylint: disable=unused-import
+                    torch._dynamo.reset() # pylint: disable=protected-access
+                    from modules.intel.openvino import openvino_fx, openvino_clear_caches # pylint: disable=unused-import
                     openvino_clear_caches()
-                    model_state.partition_id = 0
+                    torch._dynamo.eval_frame.check_if_dynamo_supported = lambda: True # pylint: disable=protected-access
                 log_level = logging.WARNING if opts.cuda_compile_verbose else logging.CRITICAL # pylint: disable=protected-access
                 if hasattr(torch, '_logging'):
                     torch._logging.set_logs(dynamo=log_level, aot=log_level, inductor=log_level) # pylint: disable=protected-access
diff --git a/modules/sd_models.py b/modules/sd_models.py
index 80a8c3339..c7a33ee21 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -818,7 +818,7 @@ def load_diffuser(checkpoint_info=None, already_loaded_state_dict=None, timer=No
                     sd_model.unet.training = False
                     sd_model.vae.training = False
                     sd_model.unet = torch.xpu.optimize(sd_model.unet, dtype=devices.dtype_unet, inplace=True, weights_prepack=False) # pylint: disable=attribute-defined-outside-init
-                    sd_model.vae = torch.xpu.optimize(sd_model.vae, dtype=devices.dtype_unet, inplace=True, weights_prepack=False) # pylint: disable=attribute-defined-outside-init
+                    sd_model.vae = torch.xpu.optimize(sd_model.vae, dtype=devices.dtype_vae, inplace=True, weights_prepack=False) # pylint: disable=attribute-defined-outside-init
                     shared.log.info("Applied IPEX Optimize.")
             except Exception as err:
                 shared.log.warning(f"IPEX Optimize not supported: {err}")