manu/optimus_v2

Usage

from transformers import AutoTokenizer, AutoModelForMaskedLM, AutoModel, AutoModelForMLM


model = AutoModelForMaskedLM.from_pretrained("manu/optimus_v2", trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained("manu/optimus_v2", trust_remote_code=True)

text = "Replace me by any text you'd like."
encoded_input = tokenizer(text, return_tensors='pt')
output = model(**encoded_input)