Skip to content

Commit 7e8a2c2

Browse files
committed
nit: rename plugin name
Signed-off-by: Mehant Kammakomati <[email protected]>
1 parent 175bfcd commit 7e8a2c2

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
@@ -5014,7 +5014,7 @@ def create_accelerator_and_postprocess(self):
50145014
# deepspeed and accelerate flags covering both trainer args and accelerate launcher
50155015
self.is_deepspeed_enabled = getattr(self.accelerator.state, "deepspeed_plugin", None) is not None
50165016
self.is_fsdp_enabled = getattr(self.accelerator.state, "fsdp_plugin", None) is not None
5017-
self.is_tp_enabled = getattr(self.accelerator.state, "tp_plugin", None) is not None
5017+
self.is_tp_enabled = getattr(self.accelerator.state, "torch_tp_plugin", None) is not None
50185018
# post accelerator creation setup
50195019
if self.is_fsdp_enabled:
50205020
fsdp_plugin = self.accelerator.state.fsdp_plugin

0 commit comments

Comments
 (0)