Thank you for your tool and provided code. I tried to train pDiff on sample data based on provided documentation on the repo, and it raised the following error during validation:
ValueError: Trying to set a tensor of shape torch.Size([77, 768]) in "weight" (which has shape torch.Size([77, 1024])), this look incorrect.
I used the same SD version indicated in the documentation (stable-diffusion-v1-5) for VAE and text encoder.
Would appreciate any guidance to resolve this issue.
Thank you for your tool and provided code. I tried to train pDiff on sample data based on provided documentation on the repo, and it raised the following error during validation:
ValueError: Trying to set a tensor of shape torch.Size([77, 768]) in "weight" (which has shape torch.Size([77, 1024])), this look incorrect.I used the same SD version indicated in the documentation (stable-diffusion-v1-5) for VAE and text encoder.
Would appreciate any guidance to resolve this issue.