File "/workspace/voltronformers/src/voltronformer/model.py", line 303, in forward
output = residual + self.attn(h, position_ids=position_ids)[0]
modules/module.py", line 1520, in _call_impl tor h/nn/modules/module.py",eline 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/workspace/voltronformers/src/voltronformer/infini_attention.py", line 109, in forward
return torch.concat(out, dim=1)
RuntimeError: torch.cat(): expected a non-empty list of Tensors