Fix model_runner output to ensure tensor conversion to NumPy.

Updated the `run_model` method to explicitly convert tensor outputs to NumPy arrays using `.numpy()`. This ensures compatibility with downstream processes relying on NumPy array inputs.
This commit is contained in:
DevTekVE
2024-12-29 17:31:27 +01:00
parent 8512406fa9
commit c59dead9fd
+1 -1
View File
@@ -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):