runtime error

e-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "/home/user/.local/lib/python3.8/site-packages/transformers/models/nat/modeling_nat.py", line 734, in forward encoder_outputs = self.encoder( File "/home/user/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "/home/user/.local/lib/python3.8/site-packages/transformers/models/nat/modeling_nat.py", line 590, in forward layer_outputs = layer_module(hidden_states, output_attentions) File "/home/user/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "/home/user/.local/lib/python3.8/site-packages/transformers/models/nat/modeling_nat.py", line 537, in forward layer_outputs = layer_module(hidden_states, output_attentions) File "/home/user/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "/home/user/.local/lib/python3.8/site-packages/transformers/models/nat/modeling_nat.py", line 480, in forward attention_outputs = self.attention(hidden_states, output_attentions=output_attentions) File "/home/user/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "/home/user/.local/lib/python3.8/site-packages/transformers/models/nat/modeling_nat.py", line 405, in forward self_outputs = self.self(hidden_states, output_attentions) File "/home/user/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "/home/user/.local/lib/python3.8/site-packages/transformers/models/nat/modeling_nat.py", line 343, in forward attention_scores = natten2dqkrpb(query_layer, key_layer, self.rpb, self.kernel_size, 1) TypeError: natten2dqkrpb() takes 4 positional arguments but 5 were given

Container logs: