Skip to content

Commit 3615939

Browse files
fix huggingface bert model name with "full name"
Signed-off-by: cliu-us <[email protected]>
1 parent 9719293 commit 3615939

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/models/conftest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1079,7 +1079,7 @@ def input_bert():
10791079
torch.FloatTensor: BERT sample input
10801080
"""
10811081
text = "Replace me by any text you'd like."
1082-
tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
1082+
tokenizer = BertTokenizer.from_pretrained("google-bert/bert-base-uncased")
10831083
return tokenizer(text, return_tensors="pt")
10841084

10851085

@@ -1091,4 +1091,4 @@ def model_bert():
10911091
Returns:
10921092
transformers.models.bert.modeling_bert.BertModel: BERT model
10931093
"""
1094-
return BertModel.from_pretrained("bert-base-uncased", torchscript=True)
1094+
return BertModel.from_pretrained("google-bert/bert-base-uncased", torchscript=True)

0 commit comments

Comments
 (0)