This is the final step1000 checkpoint from the 2k-start Talkie YaRN 32k run.
The recommended checkpoint from this run is the earlier step500 export:
xlr8harder/talkie-1930-13b-yarn-32k-tf.
The model uses a 16x YaRN extension from the 2,048-token reference config and
was continued-pretrained at 32,768 tokens on
xlr8harder/talkie-yarn-32k-gutenberg-pre1931-265m.
Training used BF16 FSDP on one 8xA100 80GB node, 8 FSDP ranks, one 32k sequence
per GPU, cosine LR decay from
1e-5 to
1e-6, 50 warmup steps, and weight
decay
0.01.
This checkpoint inherits the upstream Talkie model license, Apache-2.0. See
LICENSE. The continued-pretraining corpus has separate dataset
provenance and licensing documented at
xlr8harder/talkie-yarn-32k-gutenberg-pre1931-265m.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "xlr8harder/talkie-1930-13b-yarn-32k-step1000-tf"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
6model = AutoModelForCausalLM.from_pretrained(
7 model_id,
8 torch_dtype="auto",
9 device_map="auto",
10 trust_remote_code=True,
11)
Scores are aggregate RULER accuracy percentages from our harness, using 100
examples per task and greedy decoding. It is unclear how much RULER
unintentionally penalizes Talkie because Talkie is intentionally limited to
pre-1931 training data while some RULER tasks involve modern entities and facts;
the effect is hard to quantify here, but it is likely non-zero.
Step1000 preserved the same 32k aggregate score as step500, but was weaker at
16k and less well rounded overall. We therefore selected step500 as the main
published checkpoint.