Note: FINALLY have a config that works (for the pretrain.jsonl dataset at least)