administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-15 19:42:08 +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],
"enable_cuda_graph": True,
}
return CudaGraphRunner(ort.InferenceSession(
model_path,
providers=providers,
provider_options=options,
), options[0]["device_id"])
return CudaGraphRunner(
ort.InferenceSession(
model_path,
providers=providers,
provider_options=options,
),
options[0]["device_id"],
)
return ONNXModelRunner(ort.InferenceSession(
model_path,
providers=providers,
provider_options=options,
))
return ONNXModelRunner(
ort.InferenceSession(
model_path,
providers=providers,
provider_options=options,
)
)