From 26313a551275d342b576e582957e68af1aac1e1b Mon Sep 17 00:00:00 2001 From: root Date: Fri, 5 Jan 2024 04:52:54 +0000 Subject: [PATCH] small formatting for pre-commit hook --- examples/benchmarks/bert/src/mosaic_bert.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/benchmarks/bert/src/mosaic_bert.py b/examples/benchmarks/bert/src/mosaic_bert.py index 9307916b..bc602ae8 100644 --- 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. `_. @@ -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. `_.