File size: 436 Bytes
1808797 |
1 2 3 4 5 6 7 8 9 |
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) |