runtime error

Exit code: 1. Reason: model.safetensors: 0%| | 0.00/64.2M [00:00<?, ?B/s] model.safetensors: 100%|█████████▉| 64.2M/64.2M [00:00<00:00, 145MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 6, in <module> model = GPT2LMHeadModel.from_pretrained("burman-ai/gpt2-chatbot") File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 279, in _wrapper return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 4399, in from_pretrained ) = cls._load_pretrained_model( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 4833, in _load_pretrained_model disk_offload_index, cpu_offload_index = _load_state_dict_into_meta_model( File "/usr/local/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 116, in decorate_context return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 824, in _load_state_dict_into_meta_model _load_parameter_into_model(model, param_name, param.to(param_device)) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 712, in _load_parameter_into_model module.load_state_dict({param_type: tensor}, strict=False, assign=True) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 2581, in load_state_dict raise RuntimeError( RuntimeError: Error(s) in loading state_dict for Conv1D: size mismatch for bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([2304]).

Container logs: