Implement shared event-trajectory reasoning backbone

2026-07-23 14:23:34 +08:00
parent 68a6a3df88
commit 06f29c0f0a
11 changed files with 1405 additions and 766 deletions
--- a/export_tquery_logits_hidden.py
+++ b/export_tquery_logits_hidden.py
@@ -205,7 +205,7 @@ def main() -> None:

    n_rows = len(landmark_dataset)
    vocab_size = int(dataset.vocab_size)
-    hidden_dim = int(getattr(model, "n_embd", cfg_get(args, cfg_model, "n_embd", 120)))
+    hidden_dim = int(model.d_model)
    logits_dtype = numpy_float_dtype(args.logits_dtype)
    hidden_dtype = numpy_float_dtype(args.hidden_dtype)