Trainium spelling fix

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.
     """
 
 

optimum/neuron/utils/argument_utils.py

@@ -37,7 +37,7 @@ def validate_arg(
     expected_value: Optional[Any] = None,
 ):
     """
-    Checks that the argument called `arg_name` in `args` has a value matching what is expected for AWS Tranium
+    Checks that the argument called `arg_name` in `args` has a value matching what is expected for AWS Trainium
     to work well. By default it will patch invalid argument values if the environment variable
     `OPTIMUM_DISABLE_ARGUMENT_PATCH` is left to `"0"` (by default) and an expected value is provided.
 
@@ -87,7 +87,7 @@ def expected_validation_function(arg):
         else:
             raise_error_msg = (
                 "Aborting training. To disable automatic failure when an argument value is inferred to be wrong for "
-                "Tranium, set the environment variable OPTIMUM_DISABLE_STRICT_MODE to 1."
+                "Trainium, set the environment variable OPTIMUM_DISABLE_STRICT_MODE to 1."
             )
             raise ValueError(f"{error_msg}\n{raise_error_msg}")
 

setup.py

@@ -97,7 +97,7 @@
         "Programming Language :: Python :: 3.9",
         "Topic :: Scientific/Engineering :: Artificial Intelligence",
     ],
-    keywords="transformers, diffusers, mixed-precision training, fine-tuning, inference, tranium, inferentia, aws",
+    keywords="transformers, diffusers, mixed-precision training, fine-tuning, inference, trainium, inferentia, aws",
     url="https://huggingface.co/hardware/aws",
     author="HuggingFace Inc. Special Ops Team",
     author_email="[email protected]",

tools/create_examples_from_transformers.py

@@ -12,7 +12,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-"""Tools that downloads 🤗 Transformers training script examples and prepares them for AWS Tranium instances."""
+"""Tools that downloads 🤗 Transformers training script examples and prepares them for AWS Trainium instances."""
 
 import re
 import shutil