Skip to content

Commit 737162d

Browse files
authored
Update src/lightning/fabric/strategies/deepspeed.py
1 parent 25e8d48 commit 737162d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/fabric/strategies/deepspeed.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,7 @@ def model(self) -> "DeepSpeedEngine":
314314
@override
315315
def setup_module_and_optimizers(
316316
self, module: Module, optimizers: list[Optimizer], scheduler: Optional[_LRScheduler] = None
317-
) -> Tuple["DeepSpeedEngine", list[Optimizer], Optional[_LRScheduler]]:
317+
) -> tuple["DeepSpeedEngine", list[Optimizer], Optional[_LRScheduler]]:
318318
"""Set up a model and multiple optimizers together, along with an optional learning rate scheduler. Currently,
319319
only a single optimizer is supported.
320320

0 commit comments

Comments
 (0)