Instructions to use caskcsg/cotmae_base_uncased with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use caskcsg/cotmae_base_uncased with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("caskcsg/cotmae_base_uncased") model = AutoModelForMaskedLM.from_pretrained("caskcsg/cotmae_base_uncased") - Notebooks
- Google Colab
- Kaggle
| pipeline_tag: sentence-similarity | |
| tags: | |
| - feature-extraction | |
| - sentence-similarity | |
| - transformers | |
| # CoT-MAE base uncased | |
| CoT-MAE is a transformers based Mask Auto-Encoder pretraining architecture designed for Dense Passage Retrieval. | |
| **CoT-MAE base uncased** is a general pre-training language model trained with unsupervised MS-Marco corpus. | |
| Details can be found in our paper and codes. | |
| Paper: [ConTextual Mask Auto-Encoder for Dense Passage Retrieval](https://arxiv.org/abs/2208.07670). | |
| Code: [caskcsg/ir/cotmae](https://github.com/caskcsg/ir/tree/main/cotmae) | |
| ## Citations | |
| If you find our work useful, please cite our paper. | |
| ```bibtex | |
| @misc{https://doi.org/10.48550/arxiv.2208.07670, | |
| doi = {10.48550/ARXIV.2208.07670}, | |
| url = {https://arxiv.org/abs/2208.07670}, | |
| author = {Wu, Xing and Ma, Guangyuan and Lin, Meng and Lin, Zijia and Wang, Zhongyuan and Hu, Songlin}, | |
| keywords = {Computation and Language (cs.CL), Artificial Intelligence (cs.AI), FOS: Computer and information sciences, FOS: Computer and information sciences}, | |
| title = {ConTextual Mask Auto-Encoder for Dense Passage Retrieval}, | |
| publisher = {arXiv}, | |
| year = {2022}, | |
| copyright = {arXiv.org perpetual, non-exclusive license} | |
| } | |
| ``` | |