message

onnx_export.py

@@ -69,8 +69,11 @@ def convert_onnx(model_id: str, task: str, folder: str) -> List:
     if isinstance(atol, dict):
         atol = atol[task.replace("-with-past", "")]
 
-    validate_model_outputs(onnx_config, model, output, onnx_outputs, atol)
-    print(f"All good, model saved at: {output}")
+    try:
+        validate_model_outputs(onnx_config, model, output, onnx_outputs, atol)
+        print(f"All good, model saved at: {output}")
+    except ValueError:
+        print(f"An error occured, but the model was saved at: {args.output.as_posix()}")
     
     n_files = len([name for name in os.listdir(folder) if os.path.isfile(os.path.join(folder, name)) and not name.startswith(".")])