runtime error

Exit code: 1. Reason: pretrained return model_class.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 277, in _wrapper return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 5048, in from_pretrained ) = cls._load_pretrained_model( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 5468, in _load_pretrained_model _error_msgs, disk_offload_index = load_shard_file(args) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 843, in load_shard_file disk_offload_index = _load_state_dict_into_meta_model( File "/usr/local/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 120, in decorate_context return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 770, 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 667, 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 2629, in load_state_dict raise RuntimeError( RuntimeError: Error(s) in loading state_dict for Embedding: size mismatch for weight: copying a param with shape torch.Size([250002, 1024]) from checkpoint, the shape in current model is torch.Size([30522, 768]). model.safetensors: 0%| | 0.00/2.25G [00:00<?, ?B/s] model.safetensors: 0%| | 9.56M/2.25G [00:01<04:01, 9.28MB/s] model.safetensors: 12%|█▏ | 263M/2.25G [00:02<00:13, 151MB/s]  model.safetensors: 37%|███▋ | 837M/2.25G [00:03<00:04, 335MB/s] model.safetensors: 59%|█████▉ | 1.33G/2.25G [00:04<00:02, 360MB/s] model.safetensors: 100%|██████████| 2.25G/2.25G [00:05<00:00, 443MB/s]

Container logs: