Views
No views yet
1
2from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
4tokenizer = AutoTokenizer.from_pretrained("yseop/FNP_T5_D2T_simple")
5
6model = AutoModelForSeq2SeqLM.from_pretrained("yseop/FNP_T5_D2T_simple")
7
8
9text = ["Group profit | valIs | $ 10 && € $10 | dTime | in 2019"]
101
2
3input_ids = tokenizer.encode(": {}".format(text), return_tensors="pt")
4p=0.72
5k=40
6
7outputs = model.generate(input_ids,
8 do_sample=True,
9 top_p=p,
10 top_k=k,
11 early_stopping=True)
12
13print(tokenizer.decode(outputs[0]))
141
2input_ids = tokenizer.encode(": {}".format(text), return_tensors="pt")
3
4outputs = model.generate(input_ids,
5 max_length=200,
6 num_beams=2, repetition_penalty=2.5,
7 top_k=50, top_p=0.98,
8 length_penalty=1.0,
9 early_stopping=True)
10
11print(tokenizer.decode(outputs[0]))
12
13