Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

OpenVINO safety check for compiled_model_state

Browse Source
This commit is contained in:
Disty0
2025-01-31 18:35:36 +03:00
parent ac1d278af4
commit 62e1826faf
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models_compile.py
+6 -3
View File
@@ -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