GoA_Chess_Small / README.md
GofAlpha's picture
Update README.md
349ccbd verified
metadata
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

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.