Trainium spelling fix

huggingface · Jan 10, 2025 · 2836232 · 2836232
1 parent 2ebca50
commit 2836232
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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.
     """