gahow/xtrain
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

export: dump_logits bin for xserv-vs-xtrain comparison

Browse Source 
xtrain-side top-k next-token logit dump (f32 forward, same model/config/ckpt
as the exporter) mirroring xserv's dump-logits, so the closed-loop check can
compare both sides numerically for the same prompt + weights.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
Gahow Wang
2026-06-15 17:36:41 +08:00
parent 18c2229b4b
commit e246c3bec2
2 changed files with 102 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
crates/xtrain-train/Cargo.toml
View File

@@ -25,3 +25,7 @@ path = "src/bin/train.rs"
[[bin]]
name = "export_safetensors"
path = "src/bin/export_safetensors.rs"
[[bin]]
name = "dump_logits"
path = "src/bin/dump_logits.rs"