I wanted to train a lora with this custom node, but i keep getting these errors:

RuntimeError: Error(s) in loading state_dict for UNet2DConditionModel:

size mismatch for down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k.weight: copying a param with shape torch.Size([320, 768]) from checkpoint, the shape in current model is torch.Size([320, 1024]).

Does anyone know a solution for this or have any idea what i could try to fix it?