We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9719293 commit 3615939Copy full SHA for 3615939
tests/models/conftest.py
@@ -1079,7 +1079,7 @@ def input_bert():
1079
torch.FloatTensor: BERT sample input
1080
"""
1081
text = "Replace me by any text you'd like."
1082
- tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
+ tokenizer = BertTokenizer.from_pretrained("google-bert/bert-base-uncased")
1083
return tokenizer(text, return_tensors="pt")
1084
1085
@@ -1091,4 +1091,4 @@ def model_bert():
1091
Returns:
1092
transformers.models.bert.modeling_bert.BertModel: BERT model
1093
1094
- return BertModel.from_pretrained("bert-base-uncased", torchscript=True)
+ return BertModel.from_pretrained("google-bert/bert-base-uncased", torchscript=True)
0 commit comments