Transformers How to use neulab/docprompting-codet5-python-doc-retriever with Transformers:
# pip install -U transformers accelerate
# 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") 
