Skip to content

Commit 0fc8241

Browse files
committed
comment out PipelineMixedPrecisionPlugin to fix conversion error
1 parent 5aaacbc commit 0fc8241

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scripts/checkpoint_converters/convert_mistral_7b_hf_to_nemo.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -178,8 +178,8 @@ def convert(args):
178178

179179
if nemo_config.get('megatron_amp_O2', False):
180180
plugins.append(MegatronHalfPrecisionPlugin(precision=plugin_precision, device='cuda', scaler=scaler))
181-
else:
182-
plugins.append(PipelineMixedPrecisionPlugin(precision=plugin_precision, device='cuda', scaler=scaler))
181+
# else:
182+
# plugins.append(PipelineMixedPrecisionPlugin(precision=plugin_precision, device='cuda', scaler=scaler))
183183

184184
if precision == 32:
185185
dtype = torch.float32

0 commit comments

Comments
 (0)