Skip to content

vela: add the 0.8B native F32 heads, scoring and span decoding - #176

Open
MrDongsls wants to merge 11 commits into
InferMatrixLab:mainfrom
MrDongsls:model/vela2-native-heads
Open

MrDongsls wants to merge 11 commits into
InferMatrixLab:mainfrom
MrDongsls:model/vela2-native-heads

Conversation

@MrDongsls

Copy link
Copy Markdown
Collaborator

Stacked on #151 + #174 + #175 — review the last commit only.

Required human contributor answers

  1. What outcome are you trying to achieve?

    Implement the Vela-owned output side of the native path: CandidateHead, both SpanHeadV2 heads, set_bias, and the full calibration/decode chain — checked against recorded reference fixtures without any GPU. (Source: issue [New Model]: Support the Vela 2.0 series #137.)

  2. What is the highest-leverage variable?

    Head-logit parity — if the F32 math doesn't match the reference to ~1e-5, every downstream answer is wrong.

  3. How will this work move that variable?

    The recorded head-input fixtures (hidden states, logits, calibrated answers) freeze both the math and the scoring chain; five cases pin the numbers.

  4. Who is the customer or user?

    Same as vela: add the 0.8B native bundle export, wire contract and token-level processing #175 — this is the second half of the native output path.

Is a PR needed? Yes — prerequisite for the executor PR.

For a bug fix: N/A.

Purpose

heads.rs (CandidateHead bilinear + scalar-MLP with erf-GELU via libm, both SpanHeadV2 instances, set_bias — plain F32), scoring.rs (temperatures, set thresholds, PII length rule, sparse gate, span_decode_r4, label aliasing, score normalization, f64 window aggregation), head-input fixture exporter, five recorded fixture sets (~15.8 MB safetensors).

Test Plan

System1-Omni Version / Commit: model/vela2-native-heads @ 7d50f571

Check Command Result
CPU tests cargo test -p omni-vela-native 27 passed
clippy cargo clippy -p omni-vela-native --all-targets -- -D warnings clean
Docs mkdocs build --strict exit 0
Opt-in head parity VELA_NATIVE_BUNDLE=… cargo test --test vela_native_heads 5/5: logit ≤3.8e-5, answer ≤7.3e-7

Test Result

All passed. Gates default 4e-5/1e-6 (env-overridable); recorded maxima 3.7e-5 / 7.3e-7.

Demo / evidence

Author-run, x86 WSL2, CPU FP32, five cases:

Case Logit drift Answer drift
example-request 3.719e-5 7.252e-7
set-threshold 6.676e-6 1.126e-7
cjk 3.624e-5 4.768e-7
score 1.669e-6 6.077e-7
long-span-windowed 1.121e-5 0.0 (exact)

Structural assertions: discrete picks, span label/text/start/end, threshold rules — all equal.

Fixture facts: ~15.8 MB total safetensors; regenerable via documented command; rows block in JSON is informational.

Not measured: GPU, performance.

Self-review

  • Full diff reviewed.
  • Architecture and scope checked.
  • Checks run and reported above.
  • Description and claims match implementation.

Mr.Dong added 5 commits October 25, 2026 02:56
Second member of the Vela 2.0 family: a pinned vela2-decoder executor
(hybrid.py) with a 10-file gate (shard index and the side-loaded broad
head hashed; the 2.02 GB shard by exact size), member-aware response
finishing (span_heads with spans, keyed by span question ids), worker
member detection from config.json model_type, and the per-member
/v1/models entry. compare_reference.py serves either member; the hybrid
oracle fixtures export under oracle-hybrid-.

Evidence (author-run, x86 WSL2, CPU FP32, no fla — the runtime's CPU
path never dispatches to it): nine-case parity vs the pinned upstream
server exact-match (four 200s incl. key order, five agreed rejections
incl. the 16,384-window 413); 0.3B nine-case regression; live worker
smoke with span_heads and per-member thresholds; byte-identical Rust
frontend forwarding; 133 CPU contract tests.
- recipe/vela/export_native.py: gate-verified export into a native
  bundle — 320 backbone tensors re-keyed backbone. -> model. with
  per-tensor dtypes preserved (204 BF16 Linear/conv1d + 116 F32), the
  37 F32 heads incl. the side-loaded broad span head in one file,
  backbone_config at the config root (the family Config::load shape),
  and a manifest pinning the source revision, the gate table, the
  tensor inventory and every bundle file's SHA-256.
- src/models/vela/native (omni-vela-native): the bundle manifest gate
  and the wire admission mirroring protocol.py and the pinned runtime's
  preamble with byte-equal 422/400 bodies.
- recipe/vela/export_token_plan.py and five frozen token plans under
  tests/vela/fixtures/: the predict mapping, the row decomposition and
  the rendered records (p_ids, blocks, readout positions, truncation,
  a windowed long-span case); example-request totals 1,041 tokens,
  matching the oracle's usage.

Checks: cargo test (15 tests, no weights, no GPU), fmt/clippy clean,
mkdocs --strict.
The Rust processing of the native path, mirrored from the pinned
runtime and checked field-for-field against the five frozen token
plans (opt-in render parity with the exported bundle's tokenizer):

- state.rs: the state -> typed-parts mapping (role grouping, field
  ranges, canonical DEV2.0 JSON) with the pinned rejection triples;
- questions.rs: the SystemOne -> predict mapping — the validation walk
  and order, Noul folded into a two-option choice (choice renders
  [ABS], the reference's abstain default), over resolution,
  threshold/head carried, the set-label collision check;
- words.rs: the GLiNER2/CJK _WORD splitter and the first-token word
  mapping;
- render.rs: the encoder (char-offset tokens, memoized), the six-rule
  span-head dispatch with the hallucination-alias label replacement,
  the truncation budget, the DEV2.0 rendering (parts pass, question
  blocks with option ends and query positions, span layout a/c with
  word positions) and the window rows (head only, as the reference);
- processing.rs: the Processor over the exported bundle.

Also: the envelope error locs are flat arrays, matching the reference
triples exactly.

Checks: cargo test (26 tests) + fmt/clippy clean + opt-in render
parity (5/5 fixtures).
The Rust processing of the native path, mirrored from the pinned
runtime and checked field-for-field against the five frozen token
plans (opt-in render parity with the exported bundle's tokenizer):

- state.rs: the state -> typed-parts mapping (role grouping, field
  ranges, canonical DEV2.0 JSON) with the pinned rejection triples;
- questions.rs: the SystemOne -> predict mapping — the validation walk
  and order, Noul folded into a two-option choice (choice renders
  [ABS], the reference's abstain default), over resolution,
  threshold/head carried, the set-label collision check;
- words.rs: the GLiNER2/CJK _WORD splitter and the first-token word
  mapping;
- render.rs: the encoder (char-offset tokens, memoized), the six-rule
  span-head dispatch with the hallucination-alias label replacement,
  the truncation budget, the DEV2.0 rendering (parts pass, question
  blocks with option ends and query positions, span layout a/c with
  word positions) and the window rows (head only, as the reference);
- processing.rs: the Processor over the exported bundle.

Also: the envelope error locs are flat arrays, matching the reference
triples exactly.

Checks: cargo test (26 tests) + fmt/clippy clean + opt-in render
parity (5/5 fixtures).
- heads.rs: CandidateHead (option-end + query hiddens; bilinear +
  scalar-MLP, [ABS] last), both SpanHeadV2 instances (router and the
  side-loaded broad head) and set_bias, loaded from the exported
  bundle's heads.safetensors — plain float32, the reference's own
  autocast-off setting (erf-based GELU via libm).
- scoring.rs: the calibration mirror — per-type temperatures, set
  thresholds, the PII length rule (log-space interp) and sparse gate,
  span thresholds by id/label set, softmax/sigmoid, span_decode_r4
  (word units, unit vote, fill, edge/URL trim), label aliasing, score
  value normalization (i/(n-1)), and the window aggregation (mean over
  covering windows, -30 uncovered).
- recipe/vela/export_head_inputs.py and the recorded head-input
  fixtures: the reference's consumed hidden states, its head logits
  and its calibrated predict answers for the five frozen cases.

Opt-in parity (VELA_NATIVE_BUNDLE): five cases, head-logit drift
<= 3.8e-5, predict-answer drift <= 7.3e-7 (the windowed long-span case
exact), discrete picks, span structures and threshold rules equal.
@chatgpt-codex-connector

chatgpt-codex-connector Bot commented Oct 25, 2026 •

Copy link
Copy Markdown

Codex Review Summary

This comment shows the latest Codex review activity on this pull request.

Review Status Commit Review trigger
📝 Code Review ⚠️ Failed 2026-10-25T06:03:47.911299Z
We couldn’t complete this review. Comment @codex review to try again.
7d50f57 PR opened
ℹ️ About Codex in GitHub

Your team has set up Codex to review pull requests in this repo. Reviews are triggered when you

  • Open a pull request for review
  • Mark a draft as ready
  • Comment "@codex review" or "@codex security review".

Codex reacts with 👀 while any review is running, comments if it has suggestions, and reacts with 👍 once all reviews finish with no findings.

Mr.Dong added 3 commits October 25, 2026 14:26
…processing

# Conflicts:
#	Cargo.toml
#	README.md
#	docs/supported-models.md
#	recipe/README.md
…reference-serving

# Conflicts:
#	README.md
#	docs/supported-models.md
#	recipe/README.md
…heads

# Conflicts:
#	Cargo.toml
#	README.md
#	docs/supported-models.md
#	recipe/README.md
@MrDongsls
MrDongsls force-pushed the model/vela2-native-heads branch from 2564709 to ba86215 Compare October 25, 2026 06:48
Mr.Dong added 3 commits October 25, 2026 18:34
…ember docs

Keep the Clef and pplx-decider rows from main alongside the Vela 0.8B
hybrid additions; the package-gate descriptions now carry the
directory-symlink refusal from main.
…ows alongside the Clef/pplx-decider additions

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant