saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix RealmModelIntegrationTest.test_inference_open_qa (#21136)

Browse source 
fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2023-01-16 15:09:52 +01:00 committed by GitHub
parent a5327c6a9a
commit a45914193a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/models/realm/test_modeling_realm.py
View file

@ -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?"