Skip to content

Commit 51d7e45

Browse files
committed
nit: rename plugin name
Signed-off-by: Mehant Kammakomati <[email protected]>
1 parent 203e262 commit 51d7e45

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5118,7 +5118,7 @@ def create_accelerator_and_postprocess(self):
51185118
# deepspeed and accelerate flags covering both trainer args and accelerate launcher
51195119
self.is_deepspeed_enabled = getattr(self.accelerator.state, "deepspeed_plugin", None) is not None
51205120
self.is_fsdp_enabled = getattr(self.accelerator.state, "fsdp_plugin", None) is not None
5121-
self.is_tp_enabled = getattr(self.accelerator.state, "tp_plugin", None) is not None
5121+
self.is_tp_enabled = getattr(self.accelerator.state, "torch_tp_plugin", None) is not None
51225122
# post accelerator creation setup
51235123
if self.is_fsdp_enabled:
51245124
fsdp_plugin = self.accelerator.state.fsdp_plugin

0 commit comments

Comments
 (0)