From f1c2978b867bacc19e95121d7a9713a10eeaaf57 Mon Sep 17 00:00:00 2001 From: Hemil Desai Date: Mon, 2 Dec 2024 15:02:50 -0800 Subject: [PATCH] fix Signed-off-by: Hemil Desai --- scripts/checkpoint_converters/convert_nemo1_to_nemo2.py | 1 - 1 file changed, 1 deletion(-) diff --git a/scripts/checkpoint_converters/convert_nemo1_to_nemo2.py b/scripts/checkpoint_converters/convert_nemo1_to_nemo2.py index 4455b65f8949..5ce814d6cff3 100644 --- a/scripts/checkpoint_converters/convert_nemo1_to_nemo2.py +++ b/scripts/checkpoint_converters/convert_nemo1_to_nemo2.py @@ -42,7 +42,6 @@ from pathlib import Path import torch -from genericpath import isdir from megatron.core.dist_checkpointing.dict_utils import dict_list_map_inplace from megatron.core.dist_checkpointing.mapping import LocalNonpersistentObject, ShardedObject from omegaconf import OmegaConf