From fad0f6491c1bc2ea23c449e0481121c73d959c08 Mon Sep 17 00:00:00 2001
From: Siming Dai <908660116@qq.com>
Date: Fri, 20 Dec 2024 14:44:29 +0800
Subject: [PATCH] Update unified_checkpoint.py

---
 paddlenlp/trainer/unified_checkpoint/unified_checkpoint.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddlenlp/trainer/unified_checkpoint/unified_checkpoint.py b/paddlenlp/trainer/unified_checkpoint/unified_checkpoint.py
index 8b7654e52ca2..3c7c1e746916 100644
--- a/paddlenlp/trainer/unified_checkpoint/unified_checkpoint.py
+++ b/paddlenlp/trainer/unified_checkpoint/unified_checkpoint.py
@@ -21,7 +21,7 @@
 
 from paddlenlp.peft import LoRAModel, PrefixModelForCausalLM
 from paddlenlp.trainer.argparser import strtobool
-from paddlenlp.trainer.utils.helper import distributed_isfile
+from paddlenlp.trainer.utils.helper import distributed_file, distributed_isfile
 from paddlenlp.transformers.model_utils import (
     PretrainedModel,
     _add_variant,