From 399d319bde8fd723bd8b8838743068814a0754aa Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Sat, 27 Sep 2025 06:48:20 -0600 Subject: [PATCH] Use env vars --- frigate/util/model.py | 23 +++++++++++------------ 1 file changed, 11 insertions(+), 12 deletions(-) diff --git a/frigate/util/model.py b/frigate/util/model.py index 7dd4351f5..76c89071d 100644 --- a/frigate/util/model.py +++ b/frigate/util/model.py @@ -366,19 +366,18 @@ def get_ort_providers( ) if os.path.exists(compiled_model_path): - providers.append(provider) - options.append( - { - "migraphx_fp16_enable": 0, - } - ) + os.environ["ORT_MIGRAPHX_LOAD_COMPILED_MODEL"] = "1" + os.environ["ORT_MIGRAPHX_LOAD_COMPILED_PATH"] = compiled_model_path else: - providers.append(provider) - options.append( - { - "migraphx_fp16_enable": 0, - } - ) + os.environ["ORT_MIGRAPHX_SAVE_COMPILED_MODEL"] = "1" + os.environ["ORT_MIGRAPHX_SAVE_COMPILED_PATH"] = compiled_model_path + + providers.append(provider) + options.append( + { + "migraphx_fp16_enable": 0, + } + ) else: providers.append(provider) options.append({})