fix typos (#1113)

pytorch · Jun 24, 2024 · 15c918d · 15c918d
1 parent f200da5
commit 15c918d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/torchtune/models/mistral/scripts/compare_mistral_classifier.py b/tests/torchtune/models/mistral/scripts/compare_mistral_classifier.py
@@ -34,7 +34,7 @@ def mistral(
  rope_base: int = 10_000,
 ) -> TransformerDecoder:
  """
- Build the decoder assoicated with the mistral model. This includes:
+ Build the decoder associated with the mistral model. This includes:
  - Token embeddings
  - num_layers number of TransformerDecoderLayer blocks
  - RMS Norm layer applied to the output of the transformer

diff --git a/torchtune/models/llama2/_component_builders.py b/torchtune/models/llama2/_component_builders.py
@@ -52,7 +52,7 @@ def llama2(
  norm_eps: float = 1e-5,
 ) -> TransformerDecoder:
  """
- Build the decoder assoicated with the Llama2 model. This includes:
+ Build the decoder associated with the Llama2 model. This includes:
  - Token embeddings
  - num_layers number of TransformerDecoderLayer blocks
  - RMS Norm layer applied to the output of the transformer

diff --git a/torchtune/models/mistral/_component_builders.py b/torchtune/models/mistral/_component_builders.py
@@ -47,7 +47,7 @@ def mistral(
  rope_base: int = 10_000,
 ) -> TransformerDecoder:
  """
- Build the decoder assoicated with the mistral model. This includes:
+ Build the decoder associated with the mistral model. This includes:
  - Token embeddings
  - num_layers number of TransformerDecoderLayer blocks
  - RMS Norm layer applied to the output of the transformer