Transformers How to use pingzhili/chemberta-mlp2x-contrastive with Transformers:
# Load model directly
from transformers import AutoTokenizer, RobertaForReactionEmbedding
tokenizer = AutoTokenizer.from_pretrained("pingzhili/chemberta-mlp2x-contrastive")
model = RobertaForReactionEmbedding.from_pretrained("pingzhili/chemberta-mlp2x-contrastive", device_map="auto") 
