Hi, can you share training/inference code for image generation? I believe that this version of repository is protein thing.