Views
No views yet
tiktoken (GPT-2).GPT class from this repository.1from model import GPT
2import tiktoken
3import torch
4
5model = GPT.from_pretrained("geoffsee/auto-g-nano-2")
6enc = tiktoken.get_encoding("gpt2")
7
8# Generate text
9prompt = "The future of AI is"
10idx = torch.tensor(enc.encode(prompt)).unsqueeze(0)
11completion = model.generate(idx, max_new_tokens=50)
12print(enc.decode(completion[0].tolist()))