From e0e09996dec3255d5d664988134d6251490f42c1 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 1 Apr 2023 12:06:14 -0500
Subject: [PATCH] fix(api): restart workers on HIP memory errors

---
 api/onnx_web/worker/worker.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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",
 ]