Skip to content

Commit 10cdeb4

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 3e9e0a9 commit 10cdeb4

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

src/lightning/fabric/strategies/deepspeed.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -389,12 +389,11 @@ def module_sharded_context(self) -> AbstractContextManager:
389389
remote_device=self.remote_device,
390390
config_dict_or_path=self.config,
391391
)
392-
else:
393-
return deepspeed.zero.Init(
394-
enabled=self.zero_stage_3,
395-
remote_device=self.remote_device,
396-
config_dict_or_path=self.config,
397-
)
392+
return deepspeed.zero.Init(
393+
enabled=self.zero_stage_3,
394+
remote_device=self.remote_device,
395+
config_dict_or_path=self.config,
396+
)
398397

399398
@override
400399
def save_checkpoint(

0 commit comments

Comments
 (0)