def METHOD_NAME(self, tensor: Tensor, module: "pl.LightningModule") -> Tensor:  # type: ignore[override]
    if self.scaler is not None:
        tensor = self.scaler.scale(tensor)
    return super().METHOD_NAME(tensor, module)