optimus_v2 / README.md
manu's picture
Update README.md
42291dc verified
|
raw
history blame contribute delete
No virus
465 Bytes
metadata
library_name: transformers
tags: []

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)