Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ef5ab72f4b538d6f9ea032ac307b75b40ceef42e"
Unverified Commit 5f7a07c0 authored by Derrick Blakely's avatar Derrick Blakely Committed by GitHub
Browse files

use return dict for rag encoder (#9363)

parent ae333d04
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/rag/modeling_rag.py
View file @ 5f7a07c0
...@@ -1437,7 +1437,7 @@ class RagTokenForGeneration(RagPreTrainedModel): ...@@ -1437,7 +1437,7 @@ class RagTokenForGeneration(RagPreTrainedModel):
batch_size = context_input_ids.shape[0] // n_docs batch_size = context_input_ids.shape[0] // n_docs
encoder = self.rag.generator.get_encoder() encoder = self.rag.generator.get_encoder()
encoder_outputs = encoder(input_ids=context_input_ids, attention_mask=context_attention_mask) encoder_outputs = encoder(input_ids=context_input_ids, attention_mask=context_attention_mask, return_dict=True)
input_ids = torch.full( input_ids = torch.full(
(batch_size * num_beams, 1), (batch_size * num_beams, 1),
......
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