The model could not be fitted if not predict xstart

Thank you so much for sharing this great work.
I try to use DiT to my dataset, but I meet some problem when I train model using predict_xstart = False. The loss down a little and seem to stuck at a hight level and won't be down, when I use predict_xstart = True the loss down quickly.