inconsistent size when I use the huggingface model

size mismatch for model.layers.0.self_attn.k_proj.weight: copying a param with shape torch.Size([256, 2048]) from checkpoint, the shape in current model is torch.Size([2048, 2048]).

there are error when I use

model = AutoModelForCausalLM.from_pretrained(path)

load https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T

can anyone help me ?