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): ...@@ -480,15 +480,12 @@ class RealmModelIntegrationTest(unittest.TestCase):
def test_inference_open_qa(self): def test_inference_open_qa(self):
from transformers.models.realm.retrieval_realm import RealmRetriever from transformers.models.realm.retrieval_realm import RealmRetriever
config = RealmConfig()
tokenizer = RealmTokenizer.from_pretrained("google/realm-orqa-nq-openqa") tokenizer = RealmTokenizer.from_pretrained("google/realm-orqa-nq-openqa")
retriever = RealmRetriever.from_pretrained("google/realm-orqa-nq-openqa") retriever = RealmRetriever.from_pretrained("google/realm-orqa-nq-openqa")
model = RealmForOpenQA.from_pretrained( model = RealmForOpenQA.from_pretrained(
"google/realm-orqa-nq-openqa", "google/realm-orqa-nq-openqa",
retriever=retriever, retriever=retriever,
config=config,
) )
question = "Who is the pioneer in modern computer science?" 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