administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-15 07:35:27 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix check

Browse Source
This commit is contained in:
Nicolas Mowen 2024-10-13 18:19:33 -06:00
parent 12a0d36602
commit a7d43cc273
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
frigate/util/model.py
View File

@ -28,10 +28,14 @@ def get_ort_providers(
providers = []
options = []
for provider in providers:
if provider == "CudaExecutionProvider":
for provider in ort.get_available_providers():
if provider == "CUDAExecutionProvider":
providers.append(provider)
options.append({})
options.append(
{
"arena_extend_strategy": "kSameAsRequested",
}
)
elif provider == "TensorrtExecutionProvider":
# TensorrtExecutionProvider uses too much memory without options to control it
pass