From 1cd7c2ec799321c2b4ced1fc4bc07191e7e55da4 Mon Sep 17 00:00:00 2001 From: jacobfulano Date: Thu, 17 Aug 2023 13:23:43 -0400 Subject: [PATCH] BertConfig typing in modeling_mosaicbert.py --- llmfoundry/models/mosaicbert/modeling_mosaicbert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llmfoundry/models/mosaicbert/modeling_mosaicbert.py b/llmfoundry/models/mosaicbert/modeling_mosaicbert.py index 3e4bf46832..e080d0fd96 100644 --- a/llmfoundry/models/mosaicbert/modeling_mosaicbert.py +++ b/llmfoundry/models/mosaicbert/modeling_mosaicbert.py @@ -92,7 +92,7 @@ class BertModel(BertPreTrainedModel): ``` """ - def __init__(self, config, add_pooling_layer=True): + def __init__(self, config: BertConfig, add_pooling_layer=True): super(BertModel, self).__init__(config) self.embeddings = BertEmbeddings(config) self.encoder = BertEncoder(config)