TypeError: 'NoneType' object is not subscriptable

[Lilly-25]

When trying to run predictions from a selfextended model im getting the above error TypeError: 'NoneType' object is not subscriptable but without applying selfextend() im not getting any error and am able to run predictions

[Mooler0410]

Hi, if possible, could you please share the scripts? We are happy to assistant you to figure out what happens