From 62e1826fafbff7f1ca5ef02f576263afaf7e030a Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 31 Jan 2025 18:35:36 +0300 Subject: [PATCH] OpenVINO safety check for compiled_model_state --- modules/sd_models_compile.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/modules/sd_models_compile.py b/modules/sd_models_compile.py index 6ac54b8ab..d29a34afb 100644 --- a/modules/sd_models_compile.py +++ b/modules/sd_models_compile.py @@ -289,9 +289,12 @@ def openvino_recompile_model(p, hires=False, refiner=False): # recompile if a pa sd_models.unload_model_weights(op='model') sd_models.reload_model_weights(op='model') """ - shared.compiled_model_state.height = compile_height - shared.compiled_model_state.width = compile_width - shared.compiled_model_state.batch_size = p.batch_size + if shared.compiled_model_state is None: + shared.log.warning("OpenVINO: Compile Model State is not found, model is not compiled!") + else: + shared.compiled_model_state.height = compile_height + shared.compiled_model_state.width = compile_width + shared.compiled_model_state.batch_size = p.batch_size def openvino_post_compile(op="base"): # delete unet after OpenVINO compile