[DEBUG
] Traceback (most recent call last):

 "/usr/lib/python2.6/site-packages/salt/output/__init__.py", line 44, in display_output
display_data = get_printout(out, opts)(data).rstrip()

 "/usr/lib/python2.6/site-packages/salt/output/grains.py", line 33, in output
encoding = grains['locale_info']['defaultencoding']KeyError: 'locale_info'
Traceback (most recent call last):


 "train_DALLE.py", line 300, in <module>

main()


 "train_DALLE.py", line 219, in main

loss = dalle(caption_tokens, images, mask=mask, return_loss=True)


 "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl

result = self.forward(*input, **kwargs)


 "/home/shared/workspace/torch_research/text-to-image/dalle-pytorch/models/model_arch.py", line 421, in forward

out = self.transformer(tokens, mask = mask)


 "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl

result = self.forward(*input, **kwargs)


 "/opt/conda/lib/python3.7/site-packages/dalle_pytorch/transformer.py", line 106, in forward

return self.layers(x, **kwargs)


 "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl

result = self.forward(*input, **kwargs)


 "/opt/conda/lib/python3.7/site-packages/dalle_pytorch/reversible.py", line 139, in forward

x = x + f(x, **f_args)


 "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl

result = self.forward(*input, **kwargs)


 "/opt/conda/lib/python3.7/site-packages/dalle_pytorch/transformer.py", line 34, in forward

return self.fn(self.norm(x), **kwargs)


 "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl

result = self.forward(*input, **kwargs)


 "/opt/conda/lib/python3.7/site-packages/dalle_pytorch/attention.py", line 314, in forward

if self.noncausal_attn_len:


 "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 779, in __getattr__

type(self).__name__, name))
torch.nn.modules.module.ModuleAttributeError: 'SparseAttention' object has no attribute 'noncausal_attn_len'
