Skip to content

Commit 5ebcaac

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 5872eb9 commit 5ebcaac

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/lightning/fabric/strategies/deepspeed.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -650,7 +650,9 @@ def _init_deepspeed_distributed(self) -> None:
650650
f"MEMBER: {self.global_rank + 1}/{self.world_size}"
651651
)
652652
self._process_group_backend = self._get_process_group_backend()
653-
deepspeed.init_distributed(self._process_group_backend, distributed_port=self.cluster_environment.main_port, timeout=self._timeout)
653+
deepspeed.init_distributed(
654+
self._process_group_backend, distributed_port=self.cluster_environment.main_port, timeout=self._timeout
655+
)
654656

655657
def _set_node_environment_variables(self) -> None:
656658
assert self.cluster_environment is not None

0 commit comments

Comments
 (0)