Views
No views yet
<extra_id_1><extra_id_2><extra_id_3><extra_id_5><extra_id_6><extra_id_7>1from transformers import T5TokenizerFast, T5ForConditionalGeneration
2
3tokenizer = T5TokenizerFast.from_pretrained("WelfCrozzo/T5-L128-belarusian")
4model = T5ForConditionalGeneration.from_pretrained("WelfCrozzo/T5-L128-belarusian")
5
6x = tokenizer.encode('<extra_id_1>да зорак праз цяжкасці', return_tensors='pt')
7
8result = model.generate(x, return_dict_in_generate=True, output_scores=True,max_length=128)
9print(tokenizer.decode(result["sequences"][0]))