Skip to content

Commit

Permalink
third commit
Browse files Browse the repository at this point in the history
  • Loading branch information
whzikaros committed Mar 4, 2023
1 parent 47c2533 commit ff1ff35
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion g2pL/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -150,7 +150,7 @@ def forward(
output_attentions,
)
attention_output = cross_attention_outputs[0]
outputs = outputs + cross_attention_outputs[1:] # add cross attentions if we output attention weights
outputs = outputs + cross_attention_outputs[1:]

layer_output = apply_chunking_to_forward(
self.feed_forward_chunk, self.chunk_size_feed_forward, self.seq_len_dim, attention_output
Expand Down

0 comments on commit ff1ff35

Please sign in to comment.