Skip to content

Commit 161241e

Browse files
authored
Update deepspeed.py
1 parent 7c829b6 commit 161241e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/pytorch/plugins/precision/deepspeed.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,9 +141,9 @@ def optimizer_step( # type: ignore[override]
141141
@override
142142
def clip_gradients(
143143
self,
144-
module: Optional[Module],
145144
optimizer: Optimizer,
146145
clip_val: Union[int, float] = 0.0,
147146
gradient_clip_algorithm: GradClipAlgorithmType = GradClipAlgorithmType.NORM,
147+
module: Optional[Module] = None,
148148
) -> None:
149149
"""DeepSpeed handles gradient clipping internally."""

0 commit comments

Comments
 (0)