Revert "comment out PipelineMixedPrecisionPlugin to fix conversion er…

…ror" This reverts commit 0fc8241.
xingyaoww · Apr 16, 2024 · d7debba · d7debba
1 parent 02a981d
commit d7debba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/checkpoint_converters/convert_mistral_7b_hf_to_nemo.py b/scripts/checkpoint_converters/convert_mistral_7b_hf_to_nemo.py
@@ -178,8 +178,8 @@ def convert(args):
 
         if nemo_config.get('megatron_amp_O2', False):
             plugins.append(MegatronHalfPrecisionPlugin(precision=plugin_precision, device='cuda', scaler=scaler))
-        # else:
-        #     plugins.append(PipelineMixedPrecisionPlugin(precision=plugin_precision, device='cuda', scaler=scaler))
+        else:
+            plugins.append(PipelineMixedPrecisionPlugin(precision=plugin_precision, device='cuda', scaler=scaler))
 
     if precision == 32:
         dtype = torch.float32