predict simplekt with custom seq_len

[kheedogg]

model_name: simplekt, emb_type: qid
  File "/workspace/pykt-toolkit/examples/wandb_predict.py", line 145, in <module>
    main(params)
  File "/workspace/pykt-toolkit/examples/wandb_predict.py", line 67, in main
    model = load_model(model_name, model_config, data_config, emb_type, save_dir)
  File "/usr/local/lib/python3.10/dist-packages/pykt_toolkit-0.0.38-py3.10.egg/pykt/models/init_model.py", line 125, in load_model
  File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 2153, in load_state_dict
    raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for simpleKT:
        size mismatch for model.position_emb.weight: copying a param with shape torch.Size([1, 512, 256]) from checkpoint, the shape in current model is torch.Size([1, 200, 256]).

When i try to run examples/wandb_predict.py for simplekt, i got an error message like above.
I trained the model with different max_seq_len and here is the solution.