mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-04-26 00:27:40 +03:00
Use env vars
This commit is contained in:
parent
f5b207904c
commit
399d319bde
@ -366,19 +366,18 @@ def get_ort_providers(
|
|||||||
)
|
)
|
||||||
|
|
||||||
if os.path.exists(compiled_model_path):
|
if os.path.exists(compiled_model_path):
|
||||||
providers.append(provider)
|
os.environ["ORT_MIGRAPHX_LOAD_COMPILED_MODEL"] = "1"
|
||||||
options.append(
|
os.environ["ORT_MIGRAPHX_LOAD_COMPILED_PATH"] = compiled_model_path
|
||||||
{
|
|
||||||
"migraphx_fp16_enable": 0,
|
|
||||||
}
|
|
||||||
)
|
|
||||||
else:
|
else:
|
||||||
providers.append(provider)
|
os.environ["ORT_MIGRAPHX_SAVE_COMPILED_MODEL"] = "1"
|
||||||
options.append(
|
os.environ["ORT_MIGRAPHX_SAVE_COMPILED_PATH"] = compiled_model_path
|
||||||
{
|
|
||||||
"migraphx_fp16_enable": 0,
|
providers.append(provider)
|
||||||
}
|
options.append(
|
||||||
)
|
{
|
||||||
|
"migraphx_fp16_enable": 0,
|
||||||
|
}
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
providers.append(provider)
|
providers.append(provider)
|
||||||
options.append({})
|
options.append({})
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user