[diffusion] fix: respect dist_timeout option (#18386)

This commit is contained in:
Mick
2026-02-07 20:56:04 +08:00
committed by GitHub
parent fddef76619
commit 31d4cd2ffd
3 changed files with 20 additions and 5 deletions

View File

@@ -85,6 +85,7 @@ class GPUWorker:
ring_degree=self.server_args.ring_degree,
sp_size=self.server_args.sp_degree,
dp_size=self.server_args.dp_size,
dist_timeout=self.server_args.dist_timeout,
)
self.pipeline = build_pipeline(self.server_args)