Skip to content

LFM2.5 audio detokenizer: convert native phase coverage into a faithful FIFO program #175

Description

@sydneyrenee

Problem

The released LFM2.5 path correctly loads and uses the checkpoint’s custom audio_detokenizer; Mimi is not wired into this model. Weight ownership and strict schema validation are correct, and paired leaves cover the operator families.

The execution still materializes broad transient state between roughly fifteen phases:

  • normalized activations and three wide intermediates;
  • terminal spectral representation;
  • separate values surrounding dense and transform seams;
  • six-row 512→1282 final projection and dense inverse transform execute serially through Accelerate while peer lanes return.

Persistent ShortConv state, GQA K/V rings, overlap-add ring, and envelope may be legitimate. Their existence alone is not a violation. The violation is materializing transient values merely because the reference module or C++ function ended.

The current detokenizer_payload_math_is_owned_by_paired_assembly test checks symbol presence/absence and Accelerate seams. It does not prove numerical faithfulness, cooperative work, register chaining, or minimal materialization.

Authoritative topology

Preserve the vendored LFM2.5 detokenizer exactly:

  1. audio-code embedding/fusion with the ×6 nearest-exact embedding repeat;
  2. the eight-layer causal LFM2 operator stack;
  3. final 1282 projection;
  4. log-magnitude/angle conversion;
  5. polar complex spectrum;
  6. inverse transform;
  7. overlap-add/envelope normalization;
  8. PCM publication.

Do not substitute Mimi, simplify trained operations, or change F32/BF16/reduction semantics.

Required program

  • Fuse row-owned residual/RMSNorm/operator/FFN chains where live sets permit.
  • Keep causal state and GQA rings persistent.
  • Treat Accelerate/AMX/vForce as explicit opaque seams with documented materialization on each side.
  • Partition independent dense rows and spectral/ring work across the fixed team.
  • Alias inverse-transform destinations into dead wide storage where legal.
  • Write normalized PCM directly into playback reservations.
  • Remove redundant terminal/spectral/intermediate planes after a liveness proof.
  • Provide faithful architecture twins or fail unsupported readiness explicitly.

Verification

Frozen reference traces must cover:

  • code embedding;
  • every operator/LFM layer boundary;
  • consecutive-frame causal/KV state;
  • final spectral values;
  • polar conversion;
  • inverse transform;
  • overlap-add/envelope state;
  • complete PCM, including tails and terminal flush.

Acceptance

  • The misleading symbol-hygiene test is renamed and supplemented with full-value tests.
  • Every retained plane has a recurrence, fan-out, barrier, continuation, or opaque-seam justification.
  • Lane-zero numerical islands are removed or explicitly justified as measured opaque seams.
  • No per-frame allocation, copy-only plane, scalar fallback, or hot per-element libm call remains.
  • Consecutive-frame state and complete PCM match authoritative traces.
  • Audio-token A→B→A conversation remains clear, deterministic, and correctly retired.
  • Mimi stays unwired from LFM2.5 but available for its future Moshi tranche.

Related

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions