runtime error

Exit code: 1. Reason: 100%|██████████| 557/557 [00:00<00:00, 4.37MB/s] vocab.txt: 0%| | 0.00/895k [00:00<?, ?B/s] vocab.txt: 100%|██████████| 895k/895k [00:00<00:00, 41.7MB/s] bpe.codes: 0%| | 0.00/1.14M [00:00<?, ?B/s] bpe.codes: 100%|██████████| 1.14M/1.14M [00:00<00:00, 43.5MB/s] tokenizer.json: 0%| | 0.00/3.13M [00:00<?, ?B/s] tokenizer.json: 100%|██████████| 3.13M/3.13M [00:00<00:00, 70.5MB/s] pytorch_model.bin: 0%| | 0.00/543M [00:00<?, ?B/s] model.safetensors: 5%|▌ | 67.0M/1.27G [00:01<00:32, 37.2MB/s] model.safetensors: 31%|███▏ | 398M/1.27G [00:02<00:05, 160MB/s]  pytorch_model.bin: 14%|█▎ | 73.5M/543M [00:01<00:11, 39.5MB/s] model.safetensors: 74%|███████▎ | 934M/1.27G [00:04<00:01, 284MB/s] model.safetensors: 100%|██████████| 1.27G/1.27G [00:04<00:00, 295MB/s] pytorch_model.bin: 63%|██████▎ | 342M/543M [00:02<00:01, 134MB/s]  pytorch_model.bin: 100%|██████████| 543M/543M [00:03<00:00, 176MB/s] model.safetensors: 0%| | 0.00/543M [00:00<?, ?B/s]Traceback (most recent call last): File "/app/app.py", line 147, in <module> model_B.load_state_dict(torch.load("model_B.pth", map_location=device)) 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 TextClassifier: size mismatch for out.weight: copying a param with shape torch.Size([3, 768]) from checkpoint, the shape in current model is torch.Size([7, 768]). size mismatch for out.bias: copying a param with shape torch.Size([3]) from checkpoint, the shape in current model is torch.Size([7]). model.safetensors: 14%|█▎ | 73.3M/543M [00:01<00:09, 47.4MB/s] model.safetensors: 100%|██████████| 543M/543M [00:01<00:00, 281MB/s]

Container logs: