sebastian-hofstaetter
/

idcm-distilbert-msmarco_doc

Transformers PyTorch

English IDCM document-retrieval knowledge-distillation Inference Endpoints

Model card Files Files and versions Community

sebastian-hofstaetter commited on May 26, 2021

Commit

4d1a03b

•

1 Parent(s): b974692

fix readme

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -252,7 +252,7 @@ class IDCM_InferenceOnly(PreTrainedModel):
         return pooled
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased") # honestly not sure if that is the best way to go, but it works :)
-model = ColBERT.from_pretrained("sebastian-hofstaetter/idcm-distilbert-msmarco_doc")
 ````
 ## Effectiveness on MSMARCO Passage & TREC Deep Learning '19

         return pooled
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased") # honestly not sure if that is the best way to go, but it works :)
+model = IDCM_InferenceOnly.from_pretrained("sebastian-hofstaetter/idcm-distilbert-msmarco_doc")
 ````
 ## Effectiveness on MSMARCO Passage & TREC Deep Learning '19