We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3e9d351 commit 95c308dCopy full SHA for 95c308d
paddlenlp/trainer/utils/ckpt_converter.py
@@ -19,16 +19,16 @@
19
from typing import List, Union
20
21
import paddle
22
-from paddle.distributed.checkpoint.load_state_dict import (
+from paddle.distributed.flex_checkpoint.dcp.load_state_dict import (
23
_load_state_dict,
24
get_rank_to_read_files,
25
)
26
-from paddle.distributed.checkpoint.metadata import (
+from paddle.distributed.flex_checkpoint.dcp.metadata import (
27
LocalTensorIndex,
28
LocalTensorMetadata,
29
Metadata,
30
31
-from paddle.distributed.checkpoint.utils import flatten_state_dict
+from paddle.distributed.flex_checkpoint.dcp.utils import flatten_state_dict
32
from paddle.distributed.fleet.utils.log_util import logger
33
34
MODEL_WEIGHT_SUFFIX = ".pdparams"
0 commit comments