Views
No views yet
E / os / *her- / mos / re- / ver- / *de- / cen / do / es- / *pri- / to / on- / de / mo- / *ra- / renE os / *her- / mos / re- / ver- / *de- / cen / do es- / *pri- / to on- / de / mo- / *ra- / ren1import torch
2from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
4model_name = "compellit/mt5-scan-gl-sg"
5
6device = "cuda" if torch.cuda.is_available() else "cpu"
7
8tokenizer = AutoTokenizer.from_pretrained(model_name)
9model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
10
11text = "E / os / *her- / mos / re- / ver- / *de- / cen / do / es- / *pri- / to / on- / de / mo- / *ra- / ren"
12
13inputs = tokenizer(text, return_tensors="pt")
14
15with torch.no_grad():
16 outputs = model.generate(
17 **inputs,
18 max_length=128,
19 num_beams=1,
20 do_sample=False
21 )
22
23print(tokenizer.decode(outputs[0], skip_special_tokens=True))