[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

@@ -266,7 +266,7 @@ class ServerArgs:
hsdp_replicate_dim: int = 1
hsdp_shard_dim: int = -1
dist_timeout: int | None = None # timeout for torch.distributed
dist_timeout: int | None = 3600 # 1 hour
pipeline_config: PipelineConfig = field(default_factory=PipelineConfig, repr=False)
@@ -592,7 +592,8 @@ class ServerArgs:
"--dist-timeout",
type=int,
default=ServerArgs.dist_timeout,
help="Set timeout for torch.distributed initialization.",
help="Timeout for torch.distributed operations in seconds. "
"Increase this value if you encounter 'Connection closed by peer' errors after the service is idle. ",
)
# Prompt text file for batch processing