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

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 https://github.com/neulab/code-bert-score

Best,
Uri

Sign up or log in to comment