How to use ffgcc/InfoCSE-bert-large with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, BertForCL tokenizer = AutoTokenizer.from_pretrained("ffgcc/InfoCSE-bert-large") model = BertForCL.from_pretrained("ffgcc/InfoCSE-bert-large", device_map="auto")
