We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3e9e0a9 commit 10cdeb4Copy full SHA for 10cdeb4
src/lightning/fabric/strategies/deepspeed.py
@@ -389,12 +389,11 @@ def module_sharded_context(self) -> AbstractContextManager:
389
remote_device=self.remote_device,
390
config_dict_or_path=self.config,
391
)
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
- )
+ return deepspeed.zero.Init(
+ enabled=self.zero_stage_3,
+ remote_device=self.remote_device,
+ config_dict_or_path=self.config,
+ )
398
399
@override
400
def save_checkpoint(
0 commit comments