microsoft / CodeBERT

CodeBERT
MIT License
2.15k stars 442 forks source link

inferencing unixcoder code-generation model #256

Open Shreeshsbhat opened 1 year ago

Shreeshsbhat commented 1 year ago

hi, i have trained the model on the custom dataset, but i am not getting how to run the model for inference. the solution provided as a script which takes a json file, but i want to pass the input as string. any help is appriciated, thanks.

guoday commented 1 year ago

If you use decoder-only mode, you can use this code. If you use encoder-decoder mode, you can use this code.