fix(api): restart workers on HIP memory errors

ssube · Apr 1, 2023 · e0e0999 · e0e0999
1 parent a1f54f0
commit e0e0999
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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",
 ]