Please add segmentation map image generation code

[EladWarshawsky]

This seems like great work, I will be trying it alongside RPG-DiffusionMaster. However, I saw that you also mention having support for segmentation maps, and did not see your code. You see I already tried MultiDiffusion, and it was hugely disappointing, but your samples seem to show good results, so I would love to try if you would add the code. Thank you