File size: 1,304 Bytes
9e51ef9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
---
license: apache-2.0
tags: [chess, reinforcement-learning, grpo]
---

# model_680m_16B — RL (GRPO) checkpoints

RL post-training trajectory for the chess pre-to-post compute-allocation study.
The pretraining base and SFT init for this model are
[`model_680m_16B`](https://huggingface.co/pavelslab-nyu/Chess-Pretrain-Models/tree/main/model_680m_16B) and
[`model_680m_16B`](https://huggingface.co/pavelslab-nyu/Chess-SFT-Models/tree/main/model_680m_16B).

| | |
|---|---|
| parameters | 680m |
| pretraining tokens | 15,994,915,000 (16B) |
| checkpoints here | 4 (steps 1200–1900) |
| checkpoints saved by the run | 25 |

## Steps

`1200`, `1400`, `1700`, `1900`

## Loading

Each `global_step_N/` folder is self-contained. The models use a custom
tokenizer (`tokenizer.py`), and the remote-code resolver ignores `subfolder=`,
so download the folder first and load the **local path**:

```python
from huggingface_hub import snapshot_download
from transformers import AutoModelForCausalLM, AutoTokenizer

step = "global_step_1900"
p = snapshot_download("Pre2Post-Chess-RL/Chess-RL-Models", allow_patterns=f"model_680m_16B/{step}/*") + f"/model_680m_16B/{step}"
model = AutoModelForCausalLM.from_pretrained(p, trust_remote_code=True)
tok = AutoTokenizer.from_pretrained(p, trust_remote_code=True)
```