fix code (PaddlePaddle#6880)

DesmonDay · Aug 31, 2023 · 3b41581 · 3b41581
1 parent 4c4ef61
commit 3b41581
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlenlp/trainer/trainer.py b/paddlenlp/trainer/trainer.py
@@ -482,7 +482,7 @@ def _load_from_checkpoint(self, resume_from_checkpoint=None):
                 if os.path.isfile(weights_file):
                     # We load the model state dict on the CPU to avoid an OOM error.
                     state_dict = paddle.load(weights_file, return_numpy=True)
-                    if (isinstance(self.model, LoRAModel) and self.lmodel.lora_config.tensor_parallel_degree > 1) or (
+                    if (isinstance(self.model, LoRAModel) and self.model.lora_config.tensor_parallel_degree > 1) or (
                         isinstance(self.model, PrefixModelForCausalLM)
                         and self.model.prefix_config.tensor_parallel_degree > 1
                     ):