| from transformers import BartTokenizer, BartModel | |
| def main(): | |
| 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) | |
| H_all = outputs.last_hidden_state # noqa | |
| print(H_all.shape) # (1, 8, 1024) | |
| if __name__ == '__main__': | |
| main() | |