Instructions to use neulab/docprompting-codet5-python-doc-retriever with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use neulab/docprompting-codet5-python-doc-retriever with Transformers:
# Load model directly from transformers import AutoTokenizer, BERTScorerForCL tokenizer = AutoTokenizer.from_pretrained("neulab/docprompting-codet5-python-doc-retriever") model = BERTScorerForCL.from_pretrained("neulab/docprompting-codet5-python-doc-retriever", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| A code doc retriever given NL intent. | |
| The model is based on salesforce/codet5-base. It is finetuned on neulab/docprompting-conala | |
| For more details, check out our [github](https://github.com/shuyanzhou/docprompting) |