diff --git a/api/onnx_web/worker/worker.py b/api/onnx_web/worker/worker.py
index 151ad4e5c..61b76f1c4 100644
--- a/api/onnx_web/worker/worker.py
+++ b/api/onnx_web/worker/worker.py
@@ -19,8 +19,9 @@
 
 MEMORY_ERRORS = [
     "Failed to allocate memory",
-    "out of memory",
+    "hipErrorOutOfMemory",
     "MIOPEN failure 7",
+    "out of memory",
 ]