Update model.py
Browse files
model.py
CHANGED
@@ -972,7 +972,7 @@ class TextDecoder(nn.Module):
|
|
972 |
seq = x
|
973 |
xa = enc[f]
|
974 |
for block in self.blocks[f]:
|
975 |
-
out = block(x=x, xa=xa, mask=None, enc=
|
976 |
|
977 |
if sequential:
|
978 |
x = seq
|
|
|
972 |
seq = x
|
973 |
xa = enc[f]
|
974 |
for block in self.blocks[f]:
|
975 |
+
out = block(x=x, xa=xa, mask=None, enc=None, layer=layer)
|
976 |
|
977 |
if sequential:
|
978 |
x = seq
|