fix: remove on shutdown hook (#40)

Signed-off-by: Frost Ming <[email protected]>
bentoml · Feb 3, 2025 · f54ff9a · f54ff9a
1 parent 3f07d59
commit f54ff9a
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/src/comfy_pack/service.py b/src/comfy_pack/service.py
@@ -129,13 +129,6 @@ def generate(
             ret = ret[-1]
         return ret
 
-    @bentoml.on_shutdown
-    def on_shutdown(self):
-        logger.info("Shutting down")
-        if not EXISTING_COMFYUI_SERVER:
-            self.watch_thread.join()
-            logger.info("Watch thread finished")
-
     @bentoml.on_deployment
     @staticmethod
     def prepare_models():