Views
No views yet
PREV: sin / *fe / nin / cre- / *en- / zas | CUR: *ten / *cen- / tos / de / al- / *ta- / res | NEXT: *che- / os / de / ri- / *que- / zas | OUTPUT: *ten / *cen- / tos / de al- / *ta- / res1import torch
2from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
4model_name = "compellit/mt5-scan-gl-cx"
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 = "PREV: sin / *fe / nin / cre- / *en- / zas | CUR: *ten / *cen- / tos / de / al- / *ta- / res | NEXT: *che- / os / de / ri- / *que- / zas | OUTPUT: "
12
13inputs = tokenizer(text, return_tensors="pt")
14
15with torch.no_grad():
16 outputs = model.generate(
17 **inputs,
18 max_length=256,
19 num_beams=1,
20 do_sample=False
21 )
22
23print(tokenizer.decode(outputs[0], skip_special_tokens=True))