lllyasviel / ControlNet

Let us control diffusion models!
Apache License 2.0
28.94k stars 2.61k forks source link

dataset include different sizes #667

Open remember00000 opened 3 months ago

remember00000 commented 3 months ago

Can the model only accept that the input images are all the same size? My data set is guaranteed to be a multiple of 64 in width and height (obtained from the canny preprocessing codes in this repository), but an error is reported: RuntimeError: stack expects each tensor to be equal size, but got [512, 896, 3] at entry 0 and [512, 1152, 3] at entry 1