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