File size: 1,326 Bytes
6503d71 349ccbd 6503d71 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | ---
license: other
tags:
- chess
- transformer
- text-generation
language:
- en
pipeline_tag: text-generation
---
# GofAlpha/GoA_Chess_Small
A decoder-only transformer language model trained on Lichess standard chess games.
Generates chess moves in standard algebraic notation (PGN).
## Model Details
| Property | Value |
|----------|-------|
| Parameters | 51,544,049 |
| Hidden size | 512 |
| Layers | 8 |
| Attention heads | 8 |
| Context length | 1024 tokens |
| Tokenizer | GPT-2 |
## Files
- `model.safetensors` — Model weights
- `config.json` — Architecture configuration
- `generation_config.json` — Default generation parameters
- `tokenizer.json`, `vocab.json`, `merges.txt` — GPT-2 tokenizer files
## Usage
```python
import torch
from safetensors.torch import load_file
from transformers import GPT2TokenizerFast
# Load tokenizer
tokenizer = GPT2TokenizerFast.from_pretrained("GofAlpha/GoA_Chess_Small")
# Load weights
state_dict = load_file("model.safetensors")
# Encode a chess opening
prompt = "1. e4 e5 2. Nf3"
input_ids = tokenizer.encode(prompt, return_tensors="pt")
```
## Generation Parameters
| Parameter | Default |
|-----------|---------|
| temperature | 0.6 |
| top_k | 10 |
| top_p | 0.95 |
| max_new_tokens | 1024 |
## License
Copyright (c) G of Alpha LLC. All rights reserved.
|