CI: Kill zombie diffusion processes in CI & minor code style fix on rotary embedding fallback (#13637)

This commit is contained in:
Lianmin Zheng
2025-11-20 05:57:01 -08:00
committed by GitHub
parent 852eb6ce2a
commit a352e833c4
5 changed files with 39 additions and 7 deletions

View File

@@ -62,7 +62,7 @@ class GPUWorker:
def init_device_and_model(self) -> None:
"""Initialize the device and load the model."""
setproctitle(f"sgl_diffusion::scheduler:{self.local_rank}")
setproctitle(f"sgl_diffusion::scheduler_TP{self.local_rank}")
torch.cuda.set_device(self.local_rank)
# Set environment variables for distributed initialization
os.environ["MASTER_ADDR"] = "localhost"