RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 6 but got size 5 for tensor number 1 in the list.

[TIANTEA]

when it runs:

model_pred = unet( inp_noisy_latents, timesteps, encoder_hidden_states, added_time_ids=added_time_ids).sample

The error is "RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 6 but got size 5 for tensor number 1 in the list."

could you please tell me how to fix this thanks

[xiangweifeng]

have you sovled it?I meet same problem.

[xiangweifeng]

Hi, its related to the size of the input image.

[Yww12]

when i set batch_size==2, i also meet the same problem. When the batch_size is set to 1, this problem was solved.

[danielvegamyhre]

@Yww12 @xiangweifeng my input images are 320x240 and I am running into the same error here. I am using a batch size of 1. How did you solve this?

[xiangweifeng]

try to multiple of 64.