From 311f8a83ced02f92640d2a9cc5b36cca0bf4ef04 Mon Sep 17 00:00:00 2001 From: Parth Mannan Date: Mon, 2 Feb 2026 17:43:27 -0800 Subject: [PATCH] Fix missing import --- megatron/core/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/megatron/core/utils.py b/megatron/core/utils.py index d7b702f25ec..dac1beb7075 100644 --- a/megatron/core/utils.py +++ b/megatron/core/utils.py @@ -46,6 +46,7 @@ from megatron.core import config from megatron.core._rank_utils import log_single_rank from megatron.core.package_info import __version__ as mcore_version +from megatron.core.packed_seq_params import PackedSeqParams try: from torch.distributed._tensor import DTensor