riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Remove labels from the RagModel example (#7560)

Browse Source
This commit is contained in:
Sylvain Gugger 2020-10-04 17:39:23 -04:00 committed by GitHub
parent 99cb924bfb
commit 95f792afb0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_rag.py
View File

@ -537,7 +537,7 @@ class RagModel(RagPreTrainedModel):
>>> input_dict = tokenizer.prepare_seq2seq_batch("How many people live in Paris?", "In Paris, there are 10 million people.", return_tensors="pt")
>>> input_ids = input_dict["input_ids"]
>>> outputs = model(input_ids=input_ids, labels=input_dict["labels"])
>>> outputs = model(input_ids=input_ids)
"""
use_cache = use_cache if use_cache is not None else self.config.use_cache