We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent acc49e3 commit 9e38510Copy full SHA for 9e38510
src/transformers/modeling_utils.py
@@ -1482,11 +1482,9 @@ def post_init(self):
1482
"""
1483
self.init_weights()
1484
self._backward_compatibility_gradient_checkpointing()
1485
- # 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
1486
if self.base_model is self:
1487
self._tp_plan = self.config.base_model_tp_plan
1488
- # If current model is a base model, attach `base_model_pp_plan` from config
1489
- if self.base_model is self:
1490
self._pp_plan = self.config.base_model_pp_plan
1491
1492
def dequantize(self):
0 commit comments