Skip to content

Commit 25e8d48

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent dfce07e commit 25e8d48

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/lightning/fabric/strategies/deepspeed.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -315,8 +315,8 @@ def model(self) -> "DeepSpeedEngine":
315315
def setup_module_and_optimizers(
316316
self, module: Module, optimizers: list[Optimizer], scheduler: Optional[_LRScheduler] = None
317317
) -> Tuple["DeepSpeedEngine", list[Optimizer], Optional[_LRScheduler]]:
318-
"""Set up a model and multiple optimizers together, along with an optional learning rate scheduler.
319-
Currently, only a single optimizer is supported.
318+
"""Set up a model and multiple optimizers together, along with an optional learning rate scheduler. Currently,
319+
only a single optimizer is supported.
320320
321321
Return:
322322
The model wrapped into a :class:`deepspeed.DeepSpeedEngine` and a list with a single

0 commit comments

Comments
 (0)