From bd8396fad81c9ce4d4174ad64dc954644a20afdf Mon Sep 17 00:00:00 2001 From: c0g Date: Sat, 14 Sep 2024 06:05:15 -0700 Subject: [PATCH] Fix typo in transformer docs (#1414) --- python/mlx/nn/layers/transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/mlx/nn/layers/transformer.py b/python/mlx/nn/layers/transformer.py index 35bafe3804..4b170e88f1 100644 --- a/python/mlx/nn/layers/transformer.py +++ b/python/mlx/nn/layers/transformer.py @@ -314,7 +314,7 @@ class Transformer(Module): norm_first (bool, optional): if ``True``, encoder and decoder layers will perform layer normalization before attention and MLP operations, otherwise after. Default: ``True``. - chekpoint (bool, optional): if ``True`` perform gradient checkpointing + checkpoint (bool, optional): if ``True`` perform gradient checkpointing to reduce the memory usage at the expense of more computation. Default: ``False``. """