Diffusion with Regional Feature

Hi! I'm super interested in trying this code out with the regional feature to retrieve small objects. Would you be able to upload or send over the code to run the diffusion with regional features?

Thanks!