Skip to content

Commit 59d93ef

Browse files
committed
update
1 parent feede35 commit 59d93ef

File tree

1 file changed

+1
-1
lines changed
  • src/lightning/pytorch/plugins/precision

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def __init__(self, precision: _PRECISION_INPUT, scaler: Optional["ShardedGradSca
6363
if scaler is not None and self.precision != "16-mixed":
6464
raise ValueError(f"`precision={precision!r}` does not use a scaler, found {scaler}.")
6565

66-
self.scaler = ShardedGradScaler() if scaler is None and precision in ("16-mixed", "16-true") else None
66+
self.scaler = ShardedGradScaler() if scaler is None and precision == "16-mixed" else None
6767
self.precision = precision
6868

6969
precision_to_type = {

0 commit comments

Comments
 (0)