From eda1699a492e88d6baa3d286381fd08319b741a0 Mon Sep 17 00:00:00 2001 From: zhanluxianshen Date: Wed, 5 Mar 2025 20:34:32 +0800 Subject: [PATCH] Delete redundancy if case in model_utils 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 763c8e6b6ef1..c33f268fbd78 100755 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -1482,11 +1482,9 @@ def post_init(self): """ 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):