Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "636feba552ab1f9a1ecdd3748e1245fa45b76d0d"
Commit 1018c333 authored by alexeib's avatar alexeib Committed by Myle Ott
Browse files

default need_attn to False

parent 89e19d42
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
fairseq/models/fairseq_model.py
View file @ 1018c333
...@@ -104,7 +104,7 @@ class FairseqModel(BaseFairseqModel): ...@@ -104,7 +104,7 @@ class FairseqModel(BaseFairseqModel):
assert isinstance(self.encoder, FairseqEncoder) assert isinstance(self.encoder, FairseqEncoder)
assert isinstance(self.decoder, FairseqDecoder) assert isinstance(self.decoder, FairseqDecoder)
def forward(self, src_tokens, src_lengths, prev_output_tokens, need_attn): def forward(self, src_tokens, src_lengths, prev_output_tokens, need_attn=False):
encoder_out = self.encoder(src_tokens, src_lengths) encoder_out = self.encoder(src_tokens, src_lengths)
decoder_out = self.decoder(prev_output_tokens, encoder_out, need_attn=need_attn) decoder_out = self.decoder(prev_output_tokens, encoder_out, need_attn=need_attn)
return decoder_out return decoder_out
......
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