From 9e385109cfaf9c2cbaf496260ba68c5b45aa6076 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B9=9B=E9=9C=B2=E5=85=88=E7=94=9F?= Date: Thu, 6 Mar 2025 19:36:11 +0800 Subject: [PATCH] Delete redundancy if case in model_utils (#36559) Signed-off-by: zhanluxianshen --- src/transformers/modeling_utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py index 45a6915211..29e55e01be 100755 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -1482,11 +1482,9 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix """ self.init_weights() self._backward_compatibility_gradient_checkpointing() - # If current model is a base model, attach `base_model_tp_plan` from config + # If current model is a base model, attach `base_model_tp_plan` and `base_model_pp_plan` from config if self.base_model is self: self._tp_plan = self.config.base_model_tp_plan - # If current model is a base model, attach `base_model_pp_plan` from config - if self.base_model is self: self._pp_plan = self.config.base_model_pp_plan def dequantize(self):