import transformers from transformers import AutoModelForCausalLM model_id = "gpt2" model = AutoModelForCausalLM.from_pretrained(model_id) text = "Hello" outputs = model.generate(text) print(outputs)