Views
No views yet
| Property | Value |
|---|---|
| Parameters | 51,544,049 |
| Hidden size | 512 |
| Layers | 8 |
| Attention heads | 8 |
| Context length | 1024 tokens |
| Tokenizer | GPT-2 |
model.safetensors — Model weightsconfig.json — Architecture configurationgeneration_config.json — Default generation parameterstokenizer.json, vocab.json, merges.txt — GPT-2 tokenizer files1import torch
2from safetensors.torch import load_file
3from transformers import GPT2TokenizerFast
4
5# Load tokenizer
6tokenizer = GPT2TokenizerFast.from_pretrained("GofAlpha/GoA_Chess_Small")
7
8# Load weights
9state_dict = load_file("model.safetensors")
10
11# Encode a chess opening
12prompt = "1. e4 e5 2. Nf3"
13input_ids = tokenizer.encode(prompt, return_tensors="pt")| Parameter | Default |
|---|---|
| temperature | 0.6 |
| top_k | 10 |
| top_p | 0.95 |
| max_new_tokens | 1024 |