fix the bug

Author: zty-king

paddlenlp/trainer/trainer_utils.py

@@ -1388,7 +1388,6 @@ def init_optimizer(optimizer, model_sharded_state_dict, state_dict_metadata):
     optimizer_state_names = [".moment1_0", ".moment2_0", ".beta1_pow_acc_0", ".beta2_pow_acc_0", ".w_0"]
     inner_opt = getattr(optimizer, "_inner_opt", None)
     static_to_struct_mapping = {}
-    model_sharded_state_dict = dict(sorted(model_sharded_state_dict.items()))
     for k, v in model_sharded_state_dict.items():
         if v.local_tensor.name not in static_to_struct_mapping:
             static_to_struct_mapping[v.local_tensor.name] = k