from transformers import GPT4Tokenizer, GPT4LMHeadModel
Load GPT-4 tokenizer and model
tokenizer = GPT4Tokenizer.from_pretrained("EleutherAI/gpt-neo-2.7B")
model = GPT4LMHeadModel.from_pretrained("EleutherAI/gpt-neo-2.7B")
Example question
question = "What is the capital of France?"
Tokenize input question
inputs = tokenizer(question, return_tensors="pt")
Generate output
outputs = model.generate(input_ids=inputs["input_ids"], max_length=50, num_return_sequences=1)
Decode and print output
answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
print("Answer:", answer)
Example command for image generation
command = "Draw a cat sitting on a tree"
Tokenize input command
inputs = tokenizer(command, return_tensors="pt")
Generate output
outputs = model.generate(input_ids=inputs["input_ids"], max_length=50, num_return_sequences=1)
Decode and print output
image_description = tokenizer.decode(outputs[0], skip_special_tokens=True)
print("Image description:", image_description)
