Skip to content

Commit

Permalink
reversed pp condition back to >=1
Browse files Browse the repository at this point in the history
  • Loading branch information
kshitijkg committed Apr 4, 2024
1 parent 05ad487 commit 18fe64a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion megatron/neox_arguments/arguments.py
Original file line number Diff line number Diff line change
Expand Up @@ -1037,7 +1037,7 @@ def calculate_derived(self):
# if we set pipe_parallel_size to 0 or 1, GPT2ModelPipe.to_sequential() is called, and we run training with
# the sequential model without the PipelineModule wrapper to avoid the overhead it incurs
self.update_value(
"is_pipe_parallel", self.pipe_parallel_size > 1 and self.num_experts == 1
"is_pipe_parallel", self.pipe_parallel_size >= 1 and self.num_experts == 1
)
if self.num_experts > 1:
assert not (
Expand Down

0 comments on commit 18fe64a

Please # to comment.