How to get the right check point?

[BeiningWu]

I tried to run your work on my decive, but i had some troubles.
I change your code as following, but the checkpoint seems to dismatch

So, could you give me the right check point so that I can run the model.
Thank you very much.

[MEGlr]

Hey,
I think you have to change the "model_channels" of unet_config in the yaml file according to the dismatch error that you are getting.