administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-16 08:05:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
Nicolas Mowen 2024-10-29 07:09:38 -06:00
parent ea7f9854a7
commit b2ed2fea4f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/util/model.py
View File

@ -47,7 +47,8 @@ def get_ort_providers(
options.append( options.append(
{ {
"arena_extend_strategy": "kSameAsRequested", "arena_extend_strategy": "kSameAsRequested",
"trt_fp16_enable": requires_fp16 and os.environ.get("USE_FP_16", "True") != "False", "trt_fp16_enable": requires_fp16
and os.environ.get("USE_FP_16", "True") != "False",
"trt_timing_cache_enable": True, "trt_timing_cache_enable": True,
"trt_engine_cache_enable": True, "trt_engine_cache_enable": True,
"trt_timing_cache_path": "/config/model_cache/tensorrt/ort", "trt_timing_cache_path": "/config/model_cache/tensorrt/ort",