[diffusion] fix: respect dist_timeout option (#18386)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user