How to finetune?

[ewre324]

Hi I am using standard Llama2 finetuning methods https://colab.research.google.com/drive/1PEQyJO1-f6j0S_XJ8DV50NkpzasXkrzd?usp=sharing

But I am getting this error
RuntimeError: The size of tensor a (2048) must match the size of tensor b (256) at non-singleton dimension 2

Can you suggest where I am going wrong?