administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-01 11:07:41 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix device ID

Browse Source
This commit is contained in:
Nicolas Mowen 2025-08-19 07:51:11 -06:00
parent f1a1a90ab5
commit b7b5a0c4c2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/util/model.py
View File

@ -284,7 +284,7 @@ def post_process_yolox(
def get_ort_providers( def get_ort_providers(
force_cpu: bool = False, device: str = "AUTO", requires_fp16: bool = False force_cpu: bool = False, device: str | None = "AUTO", requires_fp16: bool = False
) -> tuple[list[str], list[dict[str, Any]]]: ) -> tuple[list[str], list[dict[str, Any]]]:
if force_cpu: if force_cpu:
return ( return (
@ -301,7 +301,7 @@ def get_ort_providers(
for provider in ort.get_available_providers(): for provider in ort.get_available_providers():
if provider == "CUDAExecutionProvider": if provider == "CUDAExecutionProvider":
device_id = 0 if not device.isdigit() else int(device) device_id = 0 if (not device or not device.isdigit()) else int(device)
providers.append(provider) providers.append(provider)
options.append( options.append(
{ {