Get error while trying to use API for this model. #2

by XiaotianHe - opened

I tried input text for both "Encryption algorithm AES [MASK_ID]" and "Encryption algorithm AES [MASK]", both return following error:

error, No mask_token (

Please advise.

error, No mask_token (mask) found on the input

urialon NeuLab @ LTI/CMU org

Hi @XiaotianHe ,
Thank you for your interest in our work!

Here is a sample input: while (<mask>) {
Make sure you have the <mask> symbol somewhere in the input, if you want to actually use it for masked token prediction.
We trained these models mostly for comparing the similarity of code snippets. For more details, see


Sign up or log in to comment