How to implement the irregular mask inpainting results

[saga1214]

Hi,

I have some questions with the irregular masking inpainting results.
How can I reproduce the results in Figure 5 at Appendix C ?
I tried to generate samples with the code in the repo, but found that some of tokens that were given in sample image turns into different tokens while being sampled from diffusion model(p_sample)

It seems that the repo does not include a inference code for inpainting results.
It would be nice if anyone can help me with inpainting results!

Thanks