Anish 83dccc20ae Restore completion-only loss masking in SFT dataloader (#582) ...

* printing steps count

* adding reply only loss for chat

* using the mask by render_conversation function of tokeniser

* undoing some changes

* putting back the comment which got removed accidently, no functionality change

2026-03-02 16:37:47 -08:00

..

base_eval.py

small touchups to the eval script, re-order items etc, cosmetic

2026-02-03 21:03:42 +00:00

base_train.py

fix minor bug in fp8 application to skip tiny matmuls

2026-02-18 23:17:29 +00:00

chat_cli.py

remove leftover mid references (#491)

2026-02-02 08:33:46 -08:00

chat_eval.py

remove leftover mid references (#491)

2026-02-02 08:33:46 -08:00

chat_rl.py

remove leftover mid references (#491)

2026-02-02 08:33:46 -08:00

chat_sft.py

Restore completion-only loss masking in SFT dataloader (#582)

2026-03-02 16:37:47 -08:00

chat_web.py

remove leftover mid references (#491)

2026-02-02 08:33:46 -08:00

tok_eval.py

initial commit

2025-10-13 06:49:24 -07:00

tok_train.py

quick fix to not OOM main speedrun script

2026-01-26 22:31:42 +00:00