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

deleted a line the broke GravesAttention
2026-09-09 11:16:00 +08:00 · 2021-05-11 00:01:42 +02:00
parent f7582107da 7ddc885f37
commit e836e04df5
2 changed files with 0 additions and 2 deletions
@@ -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:
@@ -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: