File size: 312 Bytes
3e1b852 |
1 2 3 4 5 6 7 |
from transformers import BartTokenizer, BartModel
tokenizer = BartTokenizer.from_pretrained('facebook/bart-large') model = BartModel.from_pretrained('facebook/bart-large')
inputs = tokenizer("Hello, my dog is cute", return_tensors="pt") outputs = model(**inputs)
last_hidden_states = outputs.last_hidden_state |