keep getting dead kernel

[dentadelta]

Keep getting dead kernel when running the below code block

for i in (1, 2):
loss = imagen(pixel_values, texts = captions, unet_number = i)

[dentadelta]

I think it's because I'm not using 256 x 256 image size so it doesnt match with the built-in T5 text embedded size

[wasyleque]

You should include debug (errors) to Your post I guess ...