Views
No views yet
1from transformers import T5ForConditionalGeneration, T5Tokenizer
2
3tokenizer = T5Tokenizer.from_pretrained("marksverdhei/t5-deshuffle")
4model = T5ForConditionalGeneration.from_pretrained("marksverdhei/t5-deshuffle")
5
6prompt = ' brown dog fox jumped lazy over quick the the '
7
8ids = tokenizer(prompt, return_tensors="pt").input_ids
9generated_tokens, = model.generate(ids)
10print(tokenizer.decode(generated_tokens, skip_special_tokens=True))