add dead_error property to engine client

PygmalionAI · Dec 27, 2024 · 80a9973 · 80a9973
1 parent 3dc8016
commit 80a9973
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/aphrodite/engine/multiprocessing/client.py b/aphrodite/engine/multiprocessing/client.py
@@ -381,6 +381,13 @@ def is_stopped(self) -> bool:
     def errored(self) -> bool:
         return self._errored_with is not None
 
+    @property
+    def dead_error(self) -> BaseException:
+        if self._errored_with is not None:
+            return ENGINE_DEAD_ERROR(self._errored_with)
+        else:
+            return ENGINE_DEAD_ERROR()
+
     async def generate(
         self,
         inputs: PromptInputs,