Update README.md
Browse files
README.md
CHANGED
|
@@ -9,6 +9,8 @@ tags:
|
|
| 9 |
|
| 10 |
A 231M parameter base model trained on 4.4B tokens of lichess games from January 2023 that ended in checkmate (filtered out games that were won because of time).
|
| 11 |
|
|
|
|
|
|
|
| 12 |
## Inference
|
| 13 |
```py
|
| 14 |
from transformers import GPT2LMHeadModel, AutoTokenizer
|
|
|
|
| 9 |
|
| 10 |
A 231M parameter base model trained on 4.4B tokens of lichess games from January 2023 that ended in checkmate (filtered out games that were won because of time).
|
| 11 |
|
| 12 |
+
Code repo is [here](https://github.com/nsarrazin/chessformer)
|
| 13 |
+
|
| 14 |
## Inference
|
| 15 |
```py
|
| 16 |
from transformers import GPT2LMHeadModel, AutoTokenizer
|