Sin2pi commited on
Commit
10beae2
·
verified ·
1 Parent(s): e9bf4ee

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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=enc, layer=layer)
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