From b6d44e395f255979bec2ca091bb40a5064bf54ca Mon Sep 17 00:00:00 2001 From: Yuren Hao Date: Tue, 14 Jul 2026 13:21:35 -0500 Subject: =?UTF-8?q?RESULT=2021:=20crown=20sealed=20=E2=80=94=2072.11M=20x?= =?UTF-8?q?=201.44B=20tokens=20fully=20BP-free=20(largest=20to=20date),=20?= =?UTF-8?q?gen=20gate=20passed=20on=20FineWeb;=20casc=5Fgen=20data-aware?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: Claude Fable 5 Claude-Session: https://claude.ai/code/session_014FAPDWQ49M5Ye3NpTndTpn --- ep_run/casc_gen.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'ep_run') diff --git a/ep_run/casc_gen.py b/ep_run/casc_gen.py index e535df8..1d88f7c 100644 --- a/ep_run/casc_gen.py +++ b/ep_run/casc_gen.py @@ -12,7 +12,10 @@ ap.add_argument('--topk', type=int, default=40) ap.add_argument('--prompt', default='Once upon a time') args = ap.parse_args() -DD = Path('/home/yurenh2/ept/ep_run/data/tinystories_bpe') +import sys as _sys, torch as _t +_ckpath = next((a for a in _sys.argv if a.endswith('.pt')), 'runs/stage1b_ep_muon_s55000.pt') +_cfg = _t.load(_ckpath, map_location='cpu', weights_only=False).get('config', {}) +DD = Path('/home/yurenh2/ept/ep_run/data') / _cfg.get('data', 'tinystories_bpe') vocab = pickle.load(open(DD / 'meta.pkl', 'rb'))['vocab_size'] from tokenizers import Tokenizer tk = Tokenizer.from_file(str(DD / 'tokenizer.json')) -- cgit v1.2.3