update files

Author: DesmonDay

paddlenlp/trainer/unified_checkpoint/async_uc_hander.py renamed to paddlenlp/trainer/unified_checkpoint/async_handler.py

@@ -33,10 +33,10 @@
     create_meta_dict,
 )
 
-__all__ = ["AsyncCheckpointHander"]
+__all__ = ["AsyncCheckpointHandler"]
 
 
-class AsyncCheckpointHander:
+class AsyncCheckpointHandler:
     def __init__(self, args):
         # Mainly for asynchronous saving.
         self.args = args

paddlenlp/trainer/unified_checkpoint/check_uc.py renamed to paddlenlp/trainer/unified_checkpoint/check_completion.py

@@ -35,7 +35,7 @@
 except:
     core = None
 
-from .uc_utils import (
+from .utils import (
     get_expected_state_dict,
     is_sharding_split_param_mode,
     select_model_weight_index,

paddlenlp/trainer/unified_checkpoint/uc_dynamic.py renamed to paddlenlp/trainer/unified_checkpoint/load_dynamic.py

@@ -45,7 +45,7 @@
     else:
         from paddlenlp.utils.safetensors import fast_safe_open as safe_open
 
-from .uc_utils import (
+from .utils import (
     FP32_MASTER,
     get_expected_state_dict,
     mapping_optimizer_tp_actions,

paddlenlp/trainer/unified_checkpoint/uc_locally_load.py renamed to paddlenlp/trainer/unified_checkpoint/load_local.py

@@ -39,8 +39,8 @@
 from paddlenlp.utils.log import logger
 from paddlenlp.utils.nested import nested_copy
 
-from .uc_sharding_v2 import load_unified_optimizer_split_param
-from .uc_utils import (
+from .sharding_split_param_utils import load_unified_optimizer_split_param
+from .utils import (
     FP32_MASTER,
     get_expected_keys,
     get_expected_state_dict,

paddlenlp/trainer/unified_checkpoint/uc_single_card.py renamed to paddlenlp/trainer/unified_checkpoint/load_save_single_card.py

@@ -46,7 +46,7 @@
 if is_safetensors_available():
     from safetensors.numpy import save_file as safe_save_file
 
-from .uc_utils import (
+from .utils import (
     FP32_MASTER,
     generate_base_static_name,
     get_expected_state_dict,

paddlenlp/trainer/unified_checkpoint/uc_sharding_v2.py renamed to paddlenlp/trainer/unified_checkpoint/sharding_split_param_utils.py

@@ -29,7 +29,7 @@
 )
 from paddlenlp.utils.nested import nested_copy
 
-from .uc_utils import (
+from .utils import (
     FP32_MASTER,
     generate_base_static_name,
     get_expected_state_dict,

paddlenlp/trainer/unified_checkpoint/unified_checkpoint.py

@@ -62,24 +62,21 @@
     else:
         from paddlenlp.utils.safetensors import fast_load_file as load_file
 
-from .async_uc_hander import AsyncCheckpointHander
-from .check_uc import check_unified_checkpoint, check_unified_optimizer
-from .uc_dynamic import (
+from .async_handler import AsyncCheckpointHandler
+from .check_completion import check_unified_checkpoint, check_unified_optimizer
+from .load_dynamic import (
     load_unified_checkpoint_dynamically,
     load_unified_optimizer_dynamically,
 )
-from .uc_locally_load import (
-    load_unified_checkpoint_locally,
-    load_unified_optimizer_locally,
-)
-from .uc_sharding_v2 import gather_splited_param_for_optimizer
-from .uc_single_card import (
+from .load_local import load_unified_checkpoint_locally, load_unified_optimizer_locally
+from .load_save_single_card import (
     load_single_card_checkpoint,
     load_single_card_optimizer,
     save_single_card_checkpoint,
     save_single_card_optimizer,
 )
-from .uc_utils import (
+from .sharding_split_param_utils import gather_splited_param_for_optimizer
+from .utils import (
     FP32_MASTER,
     UnifiedCheckpointOption,
     filter_params,
@@ -103,7 +100,7 @@
 class UnifiedCheckpointHandler:
     def __init__(self, args):
         self.args = args
-        self.async_handler = AsyncCheckpointHander(args)
+        self.async_handler = AsyncCheckpointHandler(args)
 
     def save_unified_checkpoint(self, model, optimizer, output_dir, signal_dir=None):
         """save unified checkpoint

paddlenlp/trainer/unified_checkpoint/uc_utils.py renamed to paddlenlp/trainer/unified_checkpoint/utils.py

@@ -18,7 +18,7 @@
 import numpy as np
 import pytest
 
-from paddlenlp.trainer.plugins.unified_checkpoint import UnifiedCheckpointOption
+from paddlenlp.trainer.unified_checkpoint.utils import UnifiedCheckpointOption
 from tests.parallel_launch import TestMultipleGpus
 from tests.testing_utils import (
     require_paddle_at_least_2_gpu,