Update model.py

model.py

@@ -969,13 +969,13 @@ class TextDecoder(nn.Module):
 
         for f in order:
             if f in enc:
-                seq = x
+
                 xa = enc[f]
                 for block in self.blocks[f]:
                     out = block(x=x, xa=xa, mask=None, enc=None, layer=layer)
 
                 if sequential:
-                    x = seq
+                    x = out
                 else:
                     a = torch.sigmoid(bln[f])
                     x = a * out + (1 - a) * x