fix queue compute node

2023-09-27 17:24:00 +08:00
parent a5e5839460
commit 01f05285c7
5 changed files with 24 additions and 19 deletions
@@ -36,7 +36,7 @@ class LocalSentenceTransformer_Text_ComputeNode(Queue_ComputeNode):
        except Exception as err:
            logger.error(f"load model {self.model} failed: {err}")
            return False
-
+        self.start()
        return True

    async def execute_task(
@@ -126,7 +126,7 @@ class LocalSentenceTransformer_Image_ComputeNode(Queue_ComputeNode):
        except Exception as err:
            logger.error(f"load model {self.model} failed: {err}")
            return False
-
+        self.start()
        return True

    def _load_image(self, source: Union[ObjectID, bytes]) -> Optional[Image]: