Views
No views yet
1from transformers import T5ForConditionalGeneration, T5Tokenizer
2
3model = T5ForConditionalGeneration.from_pretrained("simonguest/echobot-t5-small-base")
4tokenizer = T5Tokenizer.from_pretrained("simonguest/echobot-t5-small-base")
5
6input_text = "Hello, world!"
7inputs = tokenizer(input_text, return_tensors="pt")
8outputs = model.generate(**inputs, max_length=128)
9print(tokenizer.decode(outputs[0], skip_special_tokens=True))
10# Output: "Hello, world!"1# Create 50 training examples like:
2# {"input": "Happy", "target": ":-)"}
3# {"input": "Sad", "target": ":-("}
4# ... then fine-tune using standard T5 fine-tuning procedures