Skip to content

Commit 95c308d

Browse files
committed
fix path
1 parent 3e9d351 commit 95c308d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

paddlenlp/trainer/utils/ckpt_converter.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,16 +19,16 @@
1919
from typing import List, Union
2020

2121
import paddle
22-
from paddle.distributed.checkpoint.load_state_dict import (
22+
from paddle.distributed.flex_checkpoint.dcp.load_state_dict import (
2323
_load_state_dict,
2424
get_rank_to_read_files,
2525
)
26-
from paddle.distributed.checkpoint.metadata import (
26+
from paddle.distributed.flex_checkpoint.dcp.metadata import (
2727
LocalTensorIndex,
2828
LocalTensorMetadata,
2929
Metadata,
3030
)
31-
from paddle.distributed.checkpoint.utils import flatten_state_dict
31+
from paddle.distributed.flex_checkpoint.dcp.utils import flatten_state_dict
3232
from paddle.distributed.fleet.utils.log_util import logger
3333

3434
MODEL_WEIGHT_SUFFIX = ".pdparams"

0 commit comments

Comments
 (0)