Remove unused import

Signed-off-by: Abhishree <[email protected]>
NVIDIA · Feb 4, 2025 · 249d49f · 249d49f
1 parent d637242
commit 249d49f
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/nemo/lightning/megatron_parallel.py b/nemo/lightning/megatron_parallel.py
@@ -188,7 +188,6 @@ def __init__(
         convert_module_fn: Optional[Callable[[ModelT], nn.Module]] = None,
     ) -> None:
         from megatron.core import parallel_state
-        from megatron.core.tensor_parallel import set_defaults_if_not_set_tensor_model_parallel_attributes
 
         _pipeline: List[nn.Module]
         if isinstance(pipeline, nn.ModuleList):