diff --git a/perceiverar/module.py b/perceiverar/module.py index d9e200a..5267d8f 100644 --- a/perceiverar/module.py +++ b/perceiverar/module.py @@ -410,7 +410,7 @@ class PerceiverARModel(nn.Module): perciever_input = torch.cat((lags, features), dim=-1) prefix, x = ( - perciever_input[:, : self.context_length - 1, ...],, + perciever_input[:, : self.context_length - 1, ...], perciever_input[:, self.context_length - 1 :, ...], )