Update unified_checkpoint.py

PaddlePaddle · Dec 20, 2024 · fad0f64 · fad0f64
1 parent bfc88b6
commit fad0f64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,