Views
No views yet
ignore_mismatched_sizes flag):1from transformers import GPT2LMHeadModel, GPT2Tokenizer
2
3repo_id = "BikoRiko/GPT-2.3"
4# Note: ignore_mismatched_sizes=True is required because the context was expanded post-training
5model = GPT2LMHeadModel.from_pretrained(repo_id, ignore_mismatched_sizes=True)
6tokenizer = GPT2Tokenizer.from_pretrained(repo_id)
7
8# Generate text
9inputs = tokenizer("The future of technology is", return_tensors='pt')
10outputs = model.generate(**inputs, max_length=150)
11print(tokenizer.decode(outputs[0]))