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