gen_data.py only generate results from constrast model

[ahnchive]

Dear Wormhole project team,

First of all, thank you so much for making all of this available. This is fascinating work!

While I was playing with your code, I noticed an issue with gen_v6.py. It appears that I could only generate results for the "model_name" labeled as "contrast_blend." This yielded only 945 images, while the total should have been 21,735. I am aware that I can obtain the complete results directly from your repository, but my aim is to replicate the process and understand the code.

I am currently in the process of debugging this issue. However, I was curious if you were already aware of this problem and could confirm whether it has been replicated on your end.

Thank you!
Young

[ahnchive]

Please disregard my previous message. I was running gen_v6.py directly and only specifying one job number, without understanding what this is doing. It appears I need to provide all job numbers corresponding to the model_names I have!