diff --git a/optimum/neuron/trainers.py b/optimum/neuron/trainers.py index 6f5f04afb..200f3d765 100755 --- a/optimum/neuron/trainers.py +++ b/optimum/neuron/trainers.py @@ -1510,13 +1510,13 @@ def save_state(self): class NeuronTrainer(_TrainerForNeuron, Trainer): """ - Trainer that is suited for performing training on AWS Tranium instances. + Trainer that is suited for performing training on AWS Trainium instances. """ class Seq2SeqNeuronTrainer(_TrainerForNeuron, Seq2SeqTrainer): """ - Seq2SeqTrainer that is suited for performing training on AWS Tranium instances. + Seq2SeqTrainer that is suited for performing training on AWS Trainium instances. """