Skip to content

Commit 311f8a8

Browse files
committed
Fix missing import
1 parent c4bea0a commit 311f8a8

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

megatron/core/utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
from megatron.core import config
4747
from megatron.core._rank_utils import log_single_rank
4848
from megatron.core.package_info import __version__ as mcore_version
49+
from megatron.core.packed_seq_params import PackedSeqParams
4950

5051
try:
5152
from torch.distributed._tensor import DTensor

0 commit comments

Comments
 (0)