runtime error

Exit code: 1. Reason: .bin: 100%|█████████▉| 54.4M/54.4M [00:06<00:00, 8.59MB/s] model_48000.safetensors: 0%| | 0.00/1.35G [00:00<?, ?B/s] model_48000.safetensors: 1%| | 10.5M/1.35G [00:01<03:36, 6.19MB/s] model_48000.safetensors: 2%|▏ | 21.0M/1.35G [00:03<03:46, 5.85MB/s] model_48000.safetensors: 2%|▏ | 27.3M/1.35G [00:05<04:14, 5.19MB/s] model_48000.safetensors: 28%|██▊ | 373M/1.35G [00:06<00:09, 102MB/s]  model_48000.safetensors: 42%|████▏ | 562M/1.35G [00:07<00:06, 124MB/s] model_48000.safetensors: 90%|████████▉ | 1.21G/1.35G [00:08<00:00, 274MB/s] model_48000.safetensors: 100%|█████████▉| 1.35G/1.35G [00:08<00:00, 153MB/s] vocab : /usr/local/lib/python3.10/site-packages/f5_tts/infer/examples/vocab.txt token : custom model : /home/user/.cache/huggingface/hub/models--erax-ai--EraX-Smile-UnixSex-F5/snapshots/ae9f7b51c08f06d0ae31e456f5e2bc921aee8da7/models/model_48000.safetensors Traceback (most recent call last): File "/home/user/app/app.py", line 80, in <module> F5TTS_ema_model = load_f5tts() File "/home/user/app/app.py", line 61, in load_f5tts return load_model(DiT, F5TTS_model_cfg, ckpt_path) File "/usr/local/lib/python3.10/site-packages/f5_tts/infer/utils_infer.py", line 269, in load_model model = load_checkpoint(model, ckpt_path, device, dtype=dtype, use_ema=use_ema) File "/usr/local/lib/python3.10/site-packages/f5_tts/infer/utils_infer.py", line 218, in load_checkpoint model.load_state_dict(checkpoint["model_state_dict"]) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 2215, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for CFM: size mismatch for transformer.text_embed.text_embed.weight: copying a param with shape torch.Size([2565, 512]) from checkpoint, the shape in current model is torch.Size([2546, 512]).

Container logs: