Model input shape doesnt match

I fixed the error by changing in_channels

in_channels=1+class_emb_size+16,

according to
https://github.com/huggingface/diffusers/issues/2967#issuecomment-1500800012
the timestep embedding gets concatenated on top of the 5 input channels.