"vscode:/vscode.git/clone" did not exist on "b8aee2e918d7ba2d5e9e80162ae26b4806873307"
Unverified Commit a4591419 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `RealmModelIntegrationTest.test_inference_open_qa` (#21136)



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent a5327c6a
......@@ -480,15 +480,12 @@ class RealmModelIntegrationTest(unittest.TestCase):
def test_inference_open_qa(self):
from transformers.models.realm.retrieval_realm import RealmRetriever
config = RealmConfig()
tokenizer = RealmTokenizer.from_pretrained("google/realm-orqa-nq-openqa")
retriever = RealmRetriever.from_pretrained("google/realm-orqa-nq-openqa")
model = RealmForOpenQA.from_pretrained(
"google/realm-orqa-nq-openqa",
retriever=retriever,
config=config,
)
question = "Who is the pioneer in modern computer science?"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment