prompt generation using gpt neo Use it in transformers :-
from transformers import AutoTokenizer, AutoModelForCausalLM from transformers import pipeline tokenizer = AutoTokenizer.from_pretrained("DrDrek/prompt_generation") model = AutoModelForCausalLM.from_pretrained("DrDrek/prompt_generation") generator = pipeline('text-generation', model=model,tokenizer=tokenizer) generator("white horse in clouds,", do_sample=True, min_length=20)