Skip to content

Commit

Permalink
fix lora (#9659)
Browse files Browse the repository at this point in the history
  • Loading branch information
lugimzzz authored Dec 19, 2024
1 parent 25415fb commit 19fcdcb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion paddlenlp/peft/lora/lora_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ def __init__(self, model, lora_config: LoRAConfig) -> None:
self.is_pipelinemodel = True
self.model._single_to_pp_mapping = None
if (self.lora_config.tensor_parallel_degree > 1 or self.is_pipelinemodel) and (
self.lora_config.lora_use_mixer or self.use_mora
self.lora_config.lora_use_mixer or self.lora_config.use_mora
):
raise NotImplementedError("lora_use_mixer or mora is not supported in tensor parallel mode.")
if self.lora_config.tensor_parallel_degree != self.model.config.tensor_parallel_degree:
Expand Down

0 comments on commit 19fcdcb

Please sign in to comment.