How to use Sethblocks/Prompter-T5 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("Sethblocks/Prompter-T5") model = AutoModelForSeq2SeqLM.from_pretrained("Sethblocks/Prompter-T5", device_map="auto")
I trained this model based on flan-t5-base to generate ChatGPT prompts. It takes up to about 3 words and generates a prompt. example prompt: "ChatGPT: story writer"
Files info
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("Sethblocks/Prompter-T5") model = AutoModelForSeq2SeqLM.from_pretrained("Sethblocks/Prompter-T5", device_map="auto")