Does the github provide any token replacing script by Roberta?

[lysummer55]

Hi,
I see in your paper, you are discussing about suing Roberta predicted token to replace the tokens while in the example notebook, it seems the code is replacing the tokens by some random token with the same POS token. Do you know which sampler/replacer/script that I should call if I want to replace the token by Roberta predictions? Is it available on your github? Thanks!