Update unified_checkpoint.py

PaddlePaddle · Dec 20, 2024 · bfc88b6 · bfc88b6
1 parent 3c0059e
commit bfc88b6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/paddlenlp/trainer/unified_checkpoint/unified_checkpoint.py b/paddlenlp/trainer/unified_checkpoint/unified_checkpoint.py
@@ -436,8 +436,7 @@ def load_unified_optimizer(self, model, optimizer, resume_from_checkpoint):
         )
         if has_merge_optimizer_safetensors:
             optimizer_index_file = os.path.join(resume_from_checkpoint, SAFE_OPTIMIZER_INDEX_NAME)
-            if distributed_isfile(optimizer_index_file):
-                distributed_file(optimizer_index_file)
+            distributed_file(optimizer_index_file)
             with open(optimizer_index_file, "r") as f:
                 index = json.loads(f.read())