RuntimeError: Error(s) in loading state_dict for UNet2DConditionModel: size mismatch for dow

RuntimeError: Error(s) in loading state_dict for UNet2DConditionModel:
size mismatch for down_blocks.0.attentions.0.proj_in.weight: copying a param with shape
torch.Size([320, 320]) from checkpoint, the shape in current model is torch.Size([320, 320, 1, 1])
my model: stable-diffusion-2-1-768v
resolution = 768
can u help m? thanks 😦
Was this page helpful?