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")
