From a374e5a574e3ba505906f094c4571aaef117ac49 Mon Sep 17 00:00:00 2001 From: An Yang Date: Thu, 10 Nov 2022 07:25:29 +0000 Subject: [PATCH] Update config.json --- config.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.json b/config.json index 67d7ff4..c105449 100644 --- a/config.json +++ b/config.json @@ -8,7 +8,7 @@ "projection_dim": 512, "text_config": { "architectures": [ - "BertForMaskedLM" + "BertModel" ], "attention_probs_dropout_prob": 0.1, "bos_token_id": 0,