Unverified Commit 95f792af authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Remove labels from the RagModel example (#7560)

parent 99cb924b
...@@ -537,7 +537,7 @@ class RagModel(RagPreTrainedModel): ...@@ -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_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"] >>> 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 use_cache = use_cache if use_cache is not None else self.config.use_cache
......
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