Views
No views yet
1from transformers import AutoTokenizer, T5ForConditionalGeneration
2
3tokenizer = AutoTokenizer.from_pretrained("ApoTro/slovak-t5-small")
4model = T5ForConditionalGeneration.from_pretrained("ApoTro/slovak-t5-small")
5
6input_ids = tokenizer("ner veta: Do druhého kola postúpili Robert Fico a Andrej Kiska s rozdielom 4,0%.", return_tensors="pt").input_ids
7labels = tokenizer("per: Robert Fico | per: Andrej Kiska", return_tensors="pt").input_ids
8
9# the forward function automatically creates the correct decoder_input_ids
10loss = model(input_ids=input_ids, labels=labels).loss
11loss.item()