diskformer
v0.3.0
Published
Run a model larger than your GPU memory in a browser tab: weights stay in OPFS, WebGPU holds only what each step needs. A disk tier plus two reference MoE engines.
Maintainers
Readme
Run a model larger than your GPU memory in a browser tab. The weights stay on disk in the browser's private storage (OPFS); WebGPU holds only what each step needs. A disk tier for in-browser inference, with two reference engines that use it.
Plain ES modules, no dependencies, no build step. Chromium with WebGPU. Weights stay in the user's own browser storage.
Install
| How | Command |
|---|---|
| npm | npm install diskformer |
| CDN | import { Gemma4MoeSsd } from 'https://cdn.jsdelivr.net/npm/[email protected]/engines/gemma4_moe_ssd.js' |
| Copy | src/, engines/ and index.js; nothing else is needed |
Try it: a 14.4 GB model in 2.5 GB of GPU memory
Open diskformer.naklitechie.com (also on a Hugging Face Space), or serve examples/chat yourself.
Pick a model and a GPU memory budget. The page downloads the model once into OPFS, resumably, then chats.
python3 -m http.server 8000 # then open http://localhost:8000/examples/chat/Measured on a MacBook M4 Pro with 24 GB, Chrome 154, on 2026-10-05:
| Model | On disk | GPU budget | GPU memory used | Decode | Output | |---|---|---|---|---|---| | Gemma 4 26B-A4B, QAT Q4_0 | 14.4 GB | 2.5 GB | 2.02 GB | 12.4 tok/s | 9/9 replies identical to llama.cpp | | Gemma 4 26B-A4B, QAT Q4_0 | 14.4 GB | 4.3 GB | 3.93 GB | 14.2 tok/s | 9/9 | | Gemma 4 26B-A4B, QAT Q4_0 | 14.4 GB | 7 GB | 6.63 GB | 17.9 tok/s | 9/9 | | Qwen3.6 35B-A3B, Q8_0 | 36.9 GB | 4 GB | 3.52 GB | 9.2 tok/s | not compared |
"Identical" means the reply text matches llama.cpp b9830's Metal path character for character, on the same GGUF. llama-server echoes Gemma's empty thought-channel marker before each reply; the gate removes that marker first. The test covers 9 conversations, greedy decoding, up to 64 tokens each. Qwen3.6 has no llama.cpp comparison: its Metal path cannot hold 36.9 GB on this Mac.
Each row is node examples/run-demo.mjs chat --budget <GB>; decode is the median over the 9 replies. A larger budget
caches more experts. Only Apple silicon is tested; discrete GPUs and 8–16 GB machines are not.
In your own page:
import { Gemma4MoeSsd } from 'diskformer/engines/gemma4_moe_ssd.js';
const engine = await Gemma4MoeSsd.load(null, { gpuBudgetBytes: 3e9, onProgress: console.log });
for await (const { text } of engine.generate([{ role: 'user', content: 'Hello' }], { maxNewTokens: 128 })) show(text);The first load downloads the GGUF from Hugging Face and copies it into OPFS in the engine's layout. Later loads take seconds.
Why
Your model does not fit on the GPU, or it fits only by crowding out everything else. Browser engines load every weight onto the GPU at load time, including the ones a step never touches. A mixture-of-experts model uses under a tenth of its weights per token. Gemma 4's per-layer embedding table is read one row per token.
diskformer keeps those weights in OPFS, read through sync access handles in workers. Only what each step needs
moves into a GPU cache. The core is the store and the residency layer: your engine keeps its own kernels and decides
what a step needs. engines/ shows two complete engines built on it.
The engines
engines/gemma4_moe_ssd.js runs Gemma 4 26B-A4B. engines/qwen35_moe_ssd.js runs Qwen3.6 35B-A3B, and
engines/qwen3_moe_ssd.js runs Qwen3-30B-A3B as their base. They are from-scratch WebGPU engines. The dense layers and
KV cache live on the GPU. The experts stream from OPFS into a GPU cache, sized by gpuBudgetBytes. A budget that
cannot hold the dense part and two tokens' experts is refused with the numbers. Decoding is greedy. The kernels are
ported from gemma4-webgpu (Apache-2.0, engines/NOTICE). The engine files are LocalMind's, kept in step by
scripts/sync-localmind.mjs.
A table on disk
RowFile holds fixed-size rows in one OPFS file with a manifest, written once and validated by a fingerprint.
RowCache keeps slots rows on the GPU, split into planes when a row's parts live in different buffers, with an
O(1) LRU and an optional GPU id→slot map that kernels can read. examples/embedding is the whole loop in one page.
Experts on disk
ingestGguf({ url, key, plan }) streams a GGUF over HTTP ranges into OPFS in your engine's layout: plan.layout says
where each tensor goes, plan.units the byte ranges to copy or split (Q8_0 and Q4_0 blocks become a value plane and an
f16-scale plane, bit-exact). It resumes from its last 1 GiB checkpoint. For a GGUF already on disk, pass a picked File
as fetch: fileFetch(file): no download, same store. ExpertStreamer (also RecordPool) keeps a fixed GPU slot pool
of expert records: ensure(layer, ids) returns pinned slots, prefetch(layer, guesses) never blocks, and uploads go
through a mapped staging ring.
measure({ path, recordBytes }) reports sequential MB/s and random record reads; OPFS speed varies by machine.
Commands
npm test # Node unit tests: record pool, row cache, GGUF parse and splits, ingest helpers
node examples/run-demo.mjs rows # headless Chrome: a 64 MB table on disk, 2,048 rows on the GPU, every byte checked
node examples/run-demo.mjs embedding # lookup → slots → a WGSL gather kernel, every output value checked
node examples/run-demo.mjs chat --budget 2.5 # downloads Gemma once (14.4 GB) to a fixed profile, replays 9 llama.cpp replies, fails on any difference
node examples/run-demo.mjs test/browser/ingest-resume.html --models <dir> --file <gguf> # abort, resume, compare bytes
node examples/run-demo.mjs test/browser/server.html --models <dir> --file <any file> # the runner's file server rejects bad requests
npm run sync # the files shared with LocalMind match their sourcesVerify it yourself
node examples/run-demo.mjs chat runs the chat page in headless Chrome under a GPU budget and replays
examples/chat/refs/gemma.json (llama.cpp's replies); any differing reply exits 1. --models <dir> serves a local
GGUF instead of downloading it; --site <url> gates a deployed copy, such as the Space.
The code runs in LocalMind on its live site. src/ (all but measure.js)
and engines/ are byte-identical to LocalMind's copies, apart from the engines' import paths.
License
MIT. engines/ contains kernels ported from Apache-2.0 code: see engines/NOTICE. Model weights are not
distributed; each model is downloaded under its own terms. Pointers:
LocalMind · CHANGELOG.md.
