[Bugfix] Fix DiffusionWorker crash on SIGINT: ensure NCCL/ZMQ cleanup on shutdown (#3872)

Signed-off-by: WU Hang <whlbx@hotmail.com>
This commit is contained in:
wuhang authored and GitHub committed 2026-05-31 00:13:23 +08:00
1 parent 2289478e3a
commit b0f3672aec
2 files changed
+48 -25

No files matched your search

@@ -60,11 +60,11 @@ class BackgroundResources:
for proc in self.processes:
if not proc.is_alive():
continue
proc.join(30)
proc.join(5)
if proc.is_alive():
logger.warning("Terminating diffusion worker %s after timeout", proc.name)
proc.terminate()
proc.join(30)
proc.join(5)
class MultiprocDiffusionExecutor(DiffusionExecutor):