drboog / Shifted_Diffusion

Code for Shifted Diffusion for Text-to-image Generation (CVPR 2023)
Creative Commons Zero v1.0 Universal
159 stars 11 forks source link

IndexError: The shape of the mask [7] at index 0 does not match the shape of the indexed tensor [8, 1536] at index 0 #14

Open liuyangyu2003 opened 7 months ago

liuyangyu2003 commented 7 months ago

hi, how can I solove this problem? Steps: 34%|█████████████████████████████████████████████████▋ | 10347/30000 [3:03:01<5:47:44, 1.06s/it, lr=1e-5, step_loss=0.158]Traceback (most recent call last): File "finetune.py", line 869, in main() File "finetune.py", line 799, in main encoder_hidden_states[clip_empty_idx] = empty_emb[clip_empty_idx] IndexError: The shape of the mask [7] at index 0 does not match the shape of the indexed tensor [8, 1536] at index 0