This commit is contained in:
Nicolas Mowen 2025-09-16 15:52:30 -06:00
parent 3ba952e55a
commit b78b7e4f14

View File

@ -369,7 +369,7 @@ class RKNNModelRunner(BaseModelRunner):
def get_optimized_runner(
model_path: str, device: str, complex_model: bool = True, **kwargs
model_path: str, device: str | None, complex_model: bool = True, **kwargs
) -> BaseModelRunner:
"""Get an optimized runner for the hardware."""
if is_rknn_compatible(model_path):