SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Delete redundancy if case in model_utils (#36559)

Browse Source 
Signed-off-by: zhanluxianshen <zhanluxianshen@163.com>
This commit is contained in:
湛露先生
2025-03-06 19:36:11 +08:00
committed by GitHub
parent acc49e390d
commit 9e385109cf
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/modeling_utils.py
View File

@@ -1482,11 +1482,9 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
""" """
self.init_weights() self.init_weights()
self._backward_compatibility_gradient_checkpointing() 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: if self.base_model is self:
self._tp_plan = self.config.base_model_tp_plan 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 self._pp_plan = self.config.base_model_pp_plan
def dequantize(self): def dequantize(self):