administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-30 10:37:42 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
Nicolas Mowen 2025-07-07 17:52:21 -06:00
parent 1e6c1bc331
commit 04fb3689e5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/detectors/plugins/tensorrt.py
View File

@ -44,6 +44,7 @@ if TRT_SUPPORT:
else: else:
return logging.DEBUG return logging.DEBUG
class TensorRTDetectorConfig(BaseDetectorConfig): class TensorRTDetectorConfig(BaseDetectorConfig):
type: Literal[DETECTOR_KEY] type: Literal[DETECTOR_KEY]
device: int = Field(default=0, title="GPU Device Index") device: int = Field(default=0, title="GPU Device Index")
@ -227,7 +228,6 @@ class TensorRtDetector(DetectionApi):
"TensorRT detector is no longer supported on amd64 system. Please use ONNX detector instead, see https://docs.frigate.video/configuration/object_detectors#onnx for more information." "TensorRT detector is no longer supported on amd64 system. Please use ONNX detector instead, see https://docs.frigate.video/configuration/object_detectors#onnx for more information."
) )
assert TRT_SUPPORT, ( assert TRT_SUPPORT, (
f"TensorRT libraries not found, {DETECTOR_KEY} detector not present" f"TensorRT libraries not found, {DETECTOR_KEY} detector not present"
) )