Results 3 issues of Zhifeng

Hello, I would like to ask. After I have processed the images of different kernel_sizes, I am going to run the training script. At this time, it prompts that the...

Based on deepspeed-stage-2 fune, the weights are shown in the figure. If I want to load the weight inference, how should I load it? ![Image](https://github.com/user-attachments/assets/8f0904eb-46f1-430f-a6db-905fd2ecc67a)

I have a question for the author. If I use the pre-trained vae and dit model, how should I modify the ModelManager? Any suggestions?