File "/home/anthony/work/third/stable-diffusion-good/Face2Diffusion/src/modules.py", line 51, in forward x = torch.cat([x,exp],-1)

mapooon / Face2Diffusion

[CVPR 2024] Face2Diffusion for Fast and Editable Face Personalization https://arxiv.org/abs/2403.05094

Other

77 stars 6 forks source link

File "/home/anthony/work/third/stable-diffusion-good/Face2Diffusion/src/modules.py", line 51, in forward x = torch.cat([x,exp],-1) #3

Closed anthonyyuan closed 7 months ago

anthonyyuan commented 8 months ago

File "/home/anthony/work/third/stable-diffusion-good/Face2Diffusion/src/modules.py", line 51, in forward x = torch.cat([x,exp],-1) ^^^^^^^^^^^^^^^^^^^^^ RuntimeError: Tensors must have same number of dimensions: got 4 and 2

mapooon commented 8 months ago

Thank you for reporting your issue. In what case did this error occur? Also, please check the shapes of x and exp. The shape of x should be (batch size, 1536) and that of exp should be (batch size, 64).