From 2c2495cc7b0e3e2942a9310f61548f40a2bc8425 Mon Sep 17 00:00:00 2001 From: Cyril Vallez Date: Fri, 14 Mar 2025 17:36:02 +0100 Subject: [PATCH] Fix post_init() code duplication (#36727) * Update modeling_utils.py * CIs --- src/transformers/modeling_utils.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py index 7ddd5193916..2898189ebb9 100755 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -1917,9 +1917,6 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix for name, module in self.named_children(): if plan := getattr(module, "_tp_plan", None): self._tp_plan.update({f"{name}.{k}": v for k, v in plan.items()}) - for name, module in self.named_children(): - if plan := getattr(module, "_tp_plan", None): - self._tp_plan.update({f"{name}.{k}": v for k, v in plan.items()}) if self._tp_plan is not None and is_torch_greater_or_equal("2.3"): for _, v in self._tp_plan.items():