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

Fix cuda graph fallback (#20039)

Browse Source
This commit is contained in:
Nicolas Mowen
2025-09-12 06:41:26 -05:00
committed by GitHub
parent ed1e3a7c9a
commit 751678c845
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frigate/detectors/plugins/onnx.py
+3 -2
View File
@@ -143,9 +143,10 @@ class ONNXDetector(DetectionApi):
try:
# Run using CUDA graphs if available
tensor_output = self._cg_runner.run(model_input_name, tensor_input)
except Exception:
logger.warning("CUDA Graphs failed, falling back to regular run")
except Exception as e:
logger.warning(f"CUDA Graphs failed, falling back to regular run: {e}")
self._cg_runner = None
tensor_output = self.model.run(None, {model_input_name: tensor_input})
else:
# Use regular run if CUDA graphs are not available
tensor_output = self.model.run(None, {model_input_name: tensor_input})