From a7d43cc273a161845f077a620167da7bba1aa093 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Sun, 13 Oct 2024 18:19:33 -0600 Subject: [PATCH] Fix check --- frigate/util/model.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/frigate/util/model.py b/frigate/util/model.py index 5e18ed828..685cd34ec 100644 --- a/frigate/util/model.py +++ b/frigate/util/model.py @@ -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