Views
No views yet
fairseq.1from transformers import RobertaTokenizer, Data2VecForSequenceClassification, Data2VecConfig
2import torch
3
4tokenizer = RobertaTokenizer.from_pretrained("roberta-large")
5config = Data2VecConfig.from_pretrained("edugp/data2vec-nlp-base")
6model = Data2VecForSequenceClassification.from_pretrained("edugp/data2vec-nlp-base", config=config)
7# Fine-tune this model
8
9inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
10outputs = model(**inputs)
11
12prediction_logits = outputs.logits