Zonos2 support - #202

Open

stephen-dwq wants to merge 24 commits into

mainfrom

Commits on Jul 31, 2026

initial zonos2; untested
stephen-dwq
committed
7d503dc
model now actually boots again, although nothing was changed?
stephen-dwq
committed
012ae0b
modifications to m* engine for end-of-stream flush and doing proper processing for vocode in zonos to avoid token-boundary audio spikes
stephen-dwq
committed
93f8bf5
trailing audio no longer dropped
stephen-dwq
committed
a6be430
zonos2 working implementation; not benchmarked or optimized
stephen-dwq
committed
2288c0f
tests
stephen-dwq
committed
02d5d43
oops pushed build to main oops oops
stephen-dwq
committed
7088bcb
attempt at batching of the mlticodebook
stephen-dwq
committed
7cc3f71
DAC streaming decoder now keeps state on device and avoids d->h->d->h->d
stephen-dwq
committed
7a22846
batching of passage to DAC
stephen-dwq
committed
7e36555
untested cuda-graph friendly modifications to MOE dispatch
stephen-dwq
committed
bc37309
test
stephen-dwq
committed
5cdb4c5
zonos2: run the multi-codebook sampler inside the captured decode graph

stephen-dwq
and
claude
committed
d8611c7
commenting cleanup
stephen-dwq
committed
c8a12e6
finally fixed the cut audio; loop stops at 1024 for KV cache; raised to 8192
stephen-dwq
committed
6f3010b
catchup to new TPCommGroup -> CommGroup name
stephen-dwq
committed
d459f26
Using the #189, remove hackiness from ZONOS2 MoE to use proper pluggable routers
stephen-dwq
committed
8d73b39

Commits on Aug 1, 2026

Commits on Aug 2, 2026

documentation update
stephen-dwq
committed
382c0bb
ruff fixes
stephen-dwq
committed
b36bbc4
more ruff fixes; maybe we should conside gettind rid of E702
stephen-dwq
committed
08c9771