train code

[Huni0318]

Hello, thank you for providing the code and checkpoints. i would like to train the model with my dataset. Can you provide detailed config? and how to get cond_image? It seems that batch_cond["cond_image"] is not defined anywhere in the provided dataset code or training script

thank you

[mingtiannihao]

Thank you for your attention. The issue you raised has been resolved, and you should be able to pull the latest version now.