Merge pull request #479 from a-froghyar/fix-inference-attention

deleted a line the broke GravesAttention
This commit is contained in:
Eren Gölge 2021-05-11 00:01:42 +02:00 committed by GitHub
commit e836e04df5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 0 additions and 2 deletions

View File

@ -463,7 +463,6 @@ class Decoder(nn.Module):
stop_tokens = []
t = 0
self._init_states(inputs)
self.attention.init_win_idx()
self.attention.init_states(inputs)
while True:
if t > 0:

View File

@ -375,7 +375,6 @@ class Decoder(nn.Module):
else:
self._init_states(inputs, mask=None, keep_states=True)
self.attention.init_win_idx()
self.attention.init_states(inputs)
outputs, stop_tokens, alignments, t = [], [], [], 0
while True: