Skip to content

Commit 0c380bf

Browse files
committed
Update NPU compilation mode
1 parent f321ce3 commit 0c380bf

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

gimpopenvino/plugins/openvino_utils/tools/model_manager.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1218,9 +1218,9 @@ def dl_sd_15_square(self, model_id, only_npu_recompilation=False):
12181218
config = None
12191219
logging.info(f"Creating NPU model for {model_name}")
12201220

1221-
if "unet_int8" in model_name or "unet_bs1" in model_name:
1222-
config = { "NPU_DPU_GROUPS" : npu_config } if npu_config is not None else None
1223-
1221+
if "unet" in model_name:
1222+
config = { "NPU_COMPILATION_MODE_PARAMS" : "performance-hint-override=latency" }
1223+
12241224

12251225
if "unet_int8" not in model_name:
12261226
model_path_fp16 = os.path.join(install_location, model_fp16, model_name + ".xml")

0 commit comments

Comments
 (0)