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: ...@@ -593,9 +593,10 @@ class ParallelConfig:
"max_parallel_loading_workers is currently " "max_parallel_loading_workers is currently "
"not supported and will be ignored." "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( 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 @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