prevent BERT weights from being downloaded twice

2019-12-06 10:11:44 +01:00
parent 5909f71028
commit 076602bdc4
1 changed files with 2 additions and 1 deletions
--- a/examples/summarization/modeling_bertabs.py
+++ b/examples/summarization/modeling_bertabs.py
@@ -158,7 +158,8 @@ class Bert(nn.Module):

    def __init__(self):
        super(Bert, self).__init__()
-        self.model = BertModel.from_pretrained("bert-base-uncased")
+        config = BertConfig.from_pretrained("bert-base-uncased")
+        self.model = BertModel(config)

    def forward(self, input_ids, attention_mask=None, token_type_ids=None, **kwargs):
        self.eval()