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

Formatting

Browse Source
This commit is contained in:
Nicolas Mowen 2025-09-18 22:24:27 -06:00
parent ae3427158c
commit 473a061d7c
1 changed files with 15 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
frigate/detectors/detection_runners.py
View File

@ -428,14 +428,19 @@ def get_optimized_runner(
**options[0], **options[0],
"enable_cuda_graph": True, "enable_cuda_graph": True,
} }
return CudaGraphRunner(ort.InferenceSession( return CudaGraphRunner(
model_path, ort.InferenceSession(
providers=providers, model_path,
provider_options=options, providers=providers,
), options[0]["device_id"]) provider_options=options,
),
options[0]["device_id"],
)
return ONNXModelRunner(ort.InferenceSession( return ONNXModelRunner(
model_path, ort.InferenceSession(
providers=providers, model_path,
provider_options=options, providers=providers,
)) provider_options=options,
)
)