Don't use rknn if device is CPU

This commit is contained in:
Nicolas Mowen 2025-10-01 15:52:58 -06:00
parent 8307fe31aa
commit 9317477543

View File

@ -424,7 +424,8 @@ def get_optimized_runner(
) -> BaseModelRunner:
"""Get an optimized runner for the hardware."""
device = device or "AUTO"
if is_rknn_compatible(model_path):
if device != "CPU" and is_rknn_compatible(model_path):
rknn_path = auto_convert_model(model_path)
if rknn_path: