Views
No views yet
1from transformers import GPT2LMHeadModel, AutoConfig
2import json
3
4config = AutoConfig.from_pretrained("LLM-course/chess-littletestmodel")
5model = GPT2LMHeadModel.from_pretrained("LLM-course/chess-littletestmodel", config=config)
6
7# Load vocab
8with open("vocab.json") as f:
9 vocab = json.load(f)