review fix

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
NVIDIA · Aug 15, 2024 · f8ecf63 · f8ecf63
1 parent 920d6e8
commit f8ecf63
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/nemo/lightning/pytorch/strategies.py b/nemo/lightning/pytorch/strategies.py
@@ -211,10 +211,7 @@ def connect(self, model: pl.LightningModule) -> None:
             self._mcore_config = _maybe_mcore_config
 
         if hasattr(self._precision_plugin, 'dtype_config'):
-            # print("self._precision_plugin.dtype_config= " + self._precision_plugin.dtype_config)
-            # quit()
             from nemo.lightning.pytorch.plugins.precision import patch_dtype_config
-
             model.config = patch_dtype_config(self._precision_plugin.dtype_config, model.config)
             model.optim.config = patch_dtype_config(self._precision_plugin.dtype_config, model.optim.config)