diff --git a/gemma/gm/nn/_transformer.py b/gemma/gm/nn/_transformer.py index 6248e940..dffdbdaf 100644 --- a/gemma/gm/nn/_transformer.py +++ b/gemma/gm/nn/_transformer.py @@ -106,7 +106,7 @@ class Transformer(nn.Module): attention_mask: kontext.Key | None = None config: _config.TransformerConfig - # Model info to specifiy the tokenizer version and default checkpoint. + # Model info to specify the tokenizer version and default checkpoint. INFO: ClassVar[ModelInfo] = ModelInfo() def __post_init__(self):