diff --git a/selfdrive/modeld/runners/model_runner.py b/selfdrive/modeld/runners/model_runner.py
index dcc11257e0..f0f5360cba 100644
--- a/selfdrive/modeld/runners/model_runner.py
+++ b/selfdrive/modeld/runners/model_runner.py
@@ -72,7 +72,7 @@ class TinyGradRunner(ModelRunner):
     return self.tensor_inputs
 
   def run_model(self, inputs: dict[str, any]) -> np.ndarray:
-    return self.model_run(**inputs).flatten()
+    return self.model_run(**inputs).numpy().flatten()
 
 
 class ONNXRunner(ModelRunner):