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.