Add more generic OOM

triton-inference-server · May 19, 2022 · b6265da · b6265da
1 parent 125ecac
commit b6265da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sagemaker_server.cc b/src/sagemaker_server.cc
@@ -451,7 +451,7 @@ SagemakerAPIServer::SageMakerMMEHandleLoadError(
       "CUDA out of memory", /* pytorch */
       "CUDA_OUT_OF_MEMORY", /* tensorflow */
       "Out of memory",      /* generic */
-  };
+      "out of memory", "MemoryError"};
 
   EVBufferAddErrorJson(req->buffer_out, load_err);