zz

I have the same confusion. I also tried running the file directly with the pre-trained DiT model provided by the author, and it worked. So I'm guessing something went wrong...

> Thank you very much. I have solved this problem, too. I made a similar mistake.