SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

prevent BERT weights from being downloaded twice

Browse Source
This commit is contained in:
Rémi Louf
2019-12-06 10:11:44 +01:00
committed by Julien Chaumond
parent 5909f71028
commit 076602bdc4
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/summarization/modeling_bertabs.py
View File

@@ -158,7 +158,8 @@ class Bert(nn.Module):
def __init__(self): def __init__(self):
super(Bert, self).__init__() 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): def forward(self, input_ids, attention_mask=None, token_type_ids=None, **kwargs):
self.eval() self.eval()