typo

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
NVIDIA · Aug 27, 2024 · 8a76d36 · 8a76d36
1 parent abb6bd6
commit 8a76d36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/collections/llm/gpt/model/test_model_import.py b/tests/collections/llm/gpt/model/test_model_import.py
@@ -40,7 +40,7 @@ def generate_twolayer_checkpoints(config_name, hf_id):
     from transformers import AutoConfig, AutoModel, AutoTokenizer
 
     config = AutoConfig.from_pretrained(hf_id, trust_remote_code=True)
-    # Redune number of layers to two.
+    # Reduce number of layers to two.
     if hasattr(config, 'num_hidden_layers'):
         print(config.num_hidden_layers)
         config.num_hidden_layers = 2