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

Update README.md

Browse source
This commit is contained in:
Patrick von Platen 2020-09-25 16:58:29 +02:00 committed by GitHub
parent cf1c88e092
commit 5ff0d6d7d0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
model_cards/facebook/rag-token-nq/README.md
View file

@ -14,12 +14,9 @@ retriever = RagRetriever.from_pretrained("facebook/rag-token-nq", index_name="ex
model = RagTokenForGeneration.from_pretrained("facebook/rag-token-nq", retriever=retriever)
input_dict = tokenizer.prepare_seq2seq_batch("How many people live in Paris?", "In Paris, there are 10 million people.", return_tensors="pt")
outputs = model(input_ids=input_dict["input_ids"], labels=input_dict["labels"])
# outputs.loss should give 76.1230
generated = model.generate(input_ids=input_dict["input_ids"])
generated_string = tokenizer.batch_decode(generated, skip_special_tokens=True)
print(tokenizer.batch_decode(generated, skip_special_tokens=True)[0])
# generated_string should give 270,000 -> not quite correct the answer, but it also only uses a dummy index
# generated_string should give 270,000,000 -> a bit too many I think
```