Skip to content

Commit

Permalink
Apply isort and black reformatting
Browse files Browse the repository at this point in the history
Signed-off-by: Victarry <[email protected]>
  • Loading branch information
Victarry committed Nov 26, 2024
1 parent 71d02fc commit beb5096
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nemo/core/optim/optimizer_with_main_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,9 +30,9 @@

try:
from megatron.core.parallel_state import (
get_expert_data_parallel_group,
get_data_parallel_group,
get_data_parallel_world_size,
get_expert_data_parallel_group,
)
from megatron.core.tensor_parallel import copy_tensor_model_parallel_attributes

Expand Down

0 comments on commit beb5096

Please sign in to comment.