Skip to content

Commit a0e298b

Browse files
committed
Update bert large model in examples/answer_questions.py.
1 parent 5909a5a commit a0e298b

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

ML-Frameworks/pytorch-aarch64/examples/answer_questions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# *******************************************************************************
2-
# Copyright 2021-2024 Arm Limited and affiliates.
2+
# Copyright 2021-2025 Arm Limited and affiliates.
33
# SPDX-License-Identifier: Apache-2.0
44
#
55
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -117,7 +117,7 @@ def main():
117117
answer = squad_records["answer"].iloc[i_record]
118118

119119
if args["bert_large"]:
120-
model_hf_path = "Graphcore/bert-large-uncased-squad"
120+
model_hf_path = "google-bert/bert-large-uncased-whole-word-masking-finetuned-squad"
121121
model_name = "BERT Large"
122122
else:
123123
model_hf_path = "distilbert-base-uncased-distilled-squad"

0 commit comments

Comments
 (0)