fix again

Author: sufubao

lightllm/server/visualserver/manager.py

@@ -191,10 +191,13 @@ async def loop_for_netio_req(self):
 
     def clean_up(self):
         for proc in getattr(self, "model_procs", []):
-            if proc.is_alive():
-                logger.info(f"Killing VIT model process {proc.pid}")
-                proc.kill()
-                proc.join(timeout=5)
+            try:
+                if proc.is_alive():
+                    logger.info(f"Killing VIT model process {proc.pid}")
+                    proc.kill()
+                    proc.join(timeout=5)
+            except (ProcessLookupError, OSError):
+                pass
 
 
 def start_visual_process(args, pipe_writer):

lightllm/server/visualserver/visual_only_manager.py

@@ -134,10 +134,13 @@ async def infer_images(self, dp_index: int, images, events):
 
     def clean_up(self):
         for proc in getattr(self, "model_procs", []):
-            if proc.is_alive():
-                logger.info(f"Killing VIT model process {proc.pid}")
-                proc.kill()
-                proc.join(timeout=5)
+            try:
+                if proc.is_alive():
+                    logger.info(f"Killing VIT model process {proc.pid}")
+                    proc.kill()
+                    proc.join(timeout=5)
+            except (ProcessLookupError, OSError):
+                pass
 
     def exposed_remote_infer_images(self, images: List[ImageItem], ref_event: threading.Event):
         try: