search

pixeli99 / SVD_Xtend

Stable Video Diffusion Training Code and Extensions.
574 stars 55 forks source link

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

Open TIANTEA opened 4 months ago

TIANTEA commented 4 months ago

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 commented 4 months ago

have you sovled it?I meet same problem.

xiangweifeng commented 4 months ago

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

Yww12 commented 3 months ago

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 commented 2 months ago

@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 commented 1 month ago

try to multiple of 64.