small formatting for pre-commit hook

mosaicml · Jan 5, 2024 · 26313a5 · 26313a5
1 parent ea82b1b
commit 26313a5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/benchmarks/bert/src/mosaic_bert.py b/examples/benchmarks/bert/src/mosaic_bert.py
@@ -31,7 +31,7 @@ def create_mosaic_bert_mlm(pretrained_model_name: str = 'bert-base-uncased',
                            tokenizer_name: Optional[str] = None,
                            gradient_checkpointing: Optional[bool] = False,
                            pretrained_checkpoint: Optional[str] = None):
-    """MosaicBERT masked language model based on |:hugging_face:| Transformers.
+    """Mosaic BERT masked language model based on |:hugging_face:| Transformers.
 
     For more information, see
     `Transformers. <https://huggingface.co/transformers/>`_.
@@ -145,7 +145,7 @@ def create_mosaic_bert_classification(
         tokenizer_name: Optional[str] = None,
         gradient_checkpointing: Optional[bool] = False,
         pretrained_checkpoint: Optional[str] = None):
-    """MosaicBERT classification model based on |:hugging_face:| Transformers.
+    """Mosaic BERT classification model based on |:hugging_face:| Transformers.
 
     For more information, see `Transformers. <https://huggingface.co/transformers/>`_.