Unverified Commit d8819c88 authored by Lucia Fang's avatar Lucia Fang Committed by GitHub
Browse files

fix assertion for single world use case (uni) (#29429)


Signed-off-by: default avatarLu Fang <fanglu@fb.com>
Co-authored-by: default avatarLucia (Lu) Fang <fanglu@meta.com>
parent de75b0bb
......@@ -593,9 +593,10 @@ class ParallelConfig:
"max_parallel_loading_workers is currently "
"not supported and will be ignored."
)
if self.distributed_executor_backend != "mp" and self.nnodes > 1:
if self.distributed_executor_backend not in ("mp", "uni") and self.nnodes > 1:
raise ValueError(
"nnodes > 1 can only be set when distributed exectuor backend is mp."
"nnodes > 1 can only be set when distributed executor "
"backend is mp or uni."
)
@property
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment