Skip to content

Commit

Permalink
Formatting, fix indent
Browse files Browse the repository at this point in the history
  • Loading branch information
loadams committed Dec 16, 2024
1 parent 4bb8f91 commit f7e8d53
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deepspeed/runtime/zero/stage_1_and_2.py
Original file line number Diff line number Diff line change
Expand Up @@ -1069,7 +1069,7 @@ def average_tensor(self, tensor):
process_group = self.dp_process_group
# count = 0
for i, param_idx_in_group, param_id in self.params_in_ipg_bucket:
param = self.bit16_groups[group_idx][param_idx_in_group]
param = self.bit16_groups[group_idx][param_idx_in_group]

process_group = self.dp_process_group

Expand Down

0 comments on commit f7e8d53

Please sign in to comment.