Spaces:
Running
Running
from transformers import AutoTokenizer, AutoModelForCausalLM | |
tokenizer = AutoTokenizer.from_pretrained("gpt2") | |
model = AutoModelForCausalLM.from_pretrained("gpt2") | |
from transformers import pipeline, set_seed | |
generator = pipeline('text-generation', model='gpt2') | |
set_seed(42) | |
generator("Hello, I'm a language model,", max_length=30, num_return_sequences=5) | |