administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-16 03:52:09 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Use env vars

Browse Source
This commit is contained in:
Nicolas Mowen 2025-09-27 06:48:20 -06:00
parent f5b207904c
commit 399d319bde
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
frigate/util/model.py
View File

@ -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({})