runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/home/user/app/app.py", line 27, in <module> model = torch.load('model_complete_18epochs.pkl',map_location=device, weights_only=False) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 1360, in load return _load( File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 1848, in _load result = unpickler.load() File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 1837, in find_class return super().find_class(mod_name, name) AttributeError: Can't get attribute 'GPT2SdpaAttention' on <module 'transformers.models.gpt2.modeling_gpt2' from '/usr/local/lib/python3.10/site-packages/transformers/models/gpt2/modeling_gpt2.py'>

Container logs: