parakeet.ts 1.0.0

package/dist/mlx/cache.js

@@ -0,0 +1,98 @@
+import { MxArray } from '@mlx-node/core';
+function s(...dims) {
+    return BigInt64Array.from(dims.map(BigInt));
+}
+/**
+ * Cache for a single Conformer layer — stores attention K/V and conv state.
+ */
+export class ConformerCache {
+    keys = null;
+    values = null;
+    conv = null;
+    offset = 0;
+    step = 256;
+    updateAndFetchKV(k, v) {
+        if (this.keys === null || this.values === null) {
+            // Allocate initial cache
+            const kShape = k.shape();
+            const batch = Number(kShape[0]);
+            const heads = Number(kShape[1]);
+            const headDim = Number(kShape[3]);
+            const initLen = Math.ceil(Number(kShape[2]) / this.step) * this.step;
+            this.keys = MxArray.zeros(s(batch, heads, initLen, headDim), null);
+            this.values = MxArray.zeros(s(batch, heads, initLen, headDim), null);
+        }
+        const newSeq = Number(k.shape()[2]);
+        // Grow if needed
+        while (this.offset + newSeq > Number(this.keys.shape()[2])) {
+            const batch = Number(this.keys.shape()[0]);
+            const heads = Number(this.keys.shape()[1]);
+            const headDim = Number(this.keys.shape()[3]);
+            const extra = MxArray.zeros(s(batch, heads, this.step, headDim), null);
+            this.keys = MxArray.concatenate(this.keys, extra, 2);
+            this.values = MxArray.concatenate(this.values, extra, 2);
+        }
+        // Write new K and V into cache at [offset : offset+newSeq]
+        // MLX doesn't have scatter_nd; we rebuild by concatenation
+        const before = this.offset;
+        const total = Number(this.keys.shape()[2]);
+        const keysBefore = this.keys.slice(BigInt64Array.from([0n, 0n, 0n, 0n]), BigInt64Array.from([this.keys.shape()[0], this.keys.shape()[1], BigInt(before), this.keys.shape()[3]]));
+        const keysAfter = this.keys.slice(BigInt64Array.from([0n, 0n, BigInt(before + newSeq), 0n]), BigInt64Array.from([this.keys.shape()[0], this.keys.shape()[1], BigInt(total), this.keys.shape()[3]]));
+        this.keys = MxArray.concatenateMany([keysBefore, k, keysAfter], 2);
+        const valsBefore = this.values.slice(BigInt64Array.from([0n, 0n, 0n, 0n]), BigInt64Array.from([this.values.shape()[0], this.values.shape()[1], BigInt(before), this.values.shape()[3]]));
+        const valsAfter = this.values.slice(BigInt64Array.from([0n, 0n, BigInt(before + newSeq), 0n]), BigInt64Array.from([this.values.shape()[0], this.values.shape()[1], BigInt(total), this.values.shape()[3]]));
+        this.values = MxArray.concatenateMany([valsBefore, v, valsAfter], 2);
+        this.offset += newSeq;
+        const cachedK = this.keys.slice(BigInt64Array.from([0n, 0n, 0n, 0n]), BigInt64Array.from([this.keys.shape()[0], this.keys.shape()[1], BigInt(this.offset), this.keys.shape()[3]]));
+        const cachedV = this.values.slice(BigInt64Array.from([0n, 0n, 0n, 0n]), BigInt64Array.from([this.values.shape()[0], this.values.shape()[1], BigInt(this.offset), this.values.shape()[3]]));
+        return [cachedK, cachedV];
+    }
+    updateAndFetchConv(x, padding) {
+        // x: [batch, seq, channels]
+        if (this.conv === null) {
+            // Pad with zeros on the left
+            const xShape = x.shape();
+            const batch = Number(xShape[0]);
+            const ch = Number(xShape[2]);
+            const pad = MxArray.zeros(s(batch, padding, ch), null);
+            this.conv = MxArray.concatenate(pad, x, 1);
+        }
+        else {
+            // Keep only the last `padding` frames plus new input
+            const convLen = Number(this.conv.shape()[1]);
+            const keep = Math.min(padding, convLen);
+            const tail = this.conv.slice(BigInt64Array.from([0n, BigInt(convLen - keep), 0n]), this.conv.shape());
+            this.conv = MxArray.concatenate(tail, x, 1);
+        }
+        return this.conv;
+    }
+}
+/**
+ * Rotating cache for streaming Conformer: drops old frames beyond keep_size.
+ */
+export class RotatingConformerCache extends ConformerCache {
+    keepSize;
+    dropSize;
+    constructor(keepSize, cacheDrop) {
+        super();
+        this.keepSize = keepSize;
+        this.dropSize = cacheDrop;
+    }
+    updateAndFetchKV(k, v) {
+        const [cachedK, cachedV] = super.updateAndFetchKV(k, v);
+        // Trim to keepSize if we've accumulated too many frames
+        const currentLen = Number(cachedK.shape()[2]);
+        if (currentLen > this.keepSize) {
+            const start = currentLen - this.keepSize;
+            const trimmedK = cachedK.slice(BigInt64Array.from([0n, 0n, BigInt(start), 0n]), cachedK.shape());
+            const trimmedV = cachedV.slice(BigInt64Array.from([0n, 0n, BigInt(start), 0n]), cachedV.shape());
+            // Update internal state
+            this.keys = trimmedK;
+            this.values = trimmedV;
+            this.offset = this.keepSize;
+            return [trimmedK, trimmedV];
+        }
+        return [cachedK, cachedV];
+    }
+}
+//# sourceMappingURL=cache.js.map

package/dist/mlx/cache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache.js","sourceRoot":"","sources":["../../src/mlx/cache.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAEzC,SAAS,CAAC,CAAC,GAAG,IAAc;IAC1B,OAAO,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;AAC9C,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,cAAc;IACzB,IAAI,GAAmB,IAAI,CAAC;IAC5B,MAAM,GAAmB,IAAI,CAAC;IAC9B,IAAI,GAAmB,IAAI,CAAC;IAC5B,MAAM,GAAG,CAAC,CAAC;IACM,IAAI,GAAG,GAAG,CAAC;IAE5B,gBAAgB,CAAC,CAAU,EAAE,CAAU;QACrC,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,IAAI,IAAI,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;YAC/C,yBAAyB;YACzB,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;YACzB,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC;YAErE,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,CAAC;YACnE,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,CAAC;QACvE,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAEpC,iBAAiB;QACjB,OAAO,IAAI,CAAC,MAAM,GAAG,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3D,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3C,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,CAAC;YACvE,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;YACrD,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QAC3D,CAAC;QAED,2DAA2D;QAC3D,2DAA2D;QAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAE3C,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAChC,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EACpC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CACvG,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAC/B,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,EACzD,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CACtG,CAAC;QACF,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC,UAAU,EAAE,CAAC,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC;QAEnE,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAClC,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EACpC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAC7G,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CACjC,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,EACzD,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAC5G,CAAC;QACF,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC,UAAU,EAAE,CAAC,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC;QAErE,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC;QAEtB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAC7B,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EACpC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAC5G,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAC/B,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EACpC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAClH,CAAC;QAEF,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC5B,CAAC;IAED,kBAAkB,CAAC,CAAU,EAAE,OAAe;QAC5C,4BAA4B;QAC5B,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACvB,6BAA6B;YAC7B,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;YACzB,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM,GAAG,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;YACvD,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,qDAAqD;YACrD,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACxC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAC1B,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,EACpD,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAClB,CAAC;YACF,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9C,CAAC;QACD,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,sBAAuB,SAAQ,cAAc;IACvC,QAAQ,CAAS;IACjB,QAAQ,CAAS;IAElC,YAAY,QAAgB,EAAE,SAAiB;QAC7C,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC;IAC5B,CAAC;IAED,gBAAgB,CAAC,CAAU,EAAE,CAAU;QACrC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,KAAK,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAExD,wDAAwD;QACxD,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9C,IAAI,UAAU,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC/B,MAAM,KAAK,GAAG,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC;YACzC,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAC5B,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,EAC/C,OAAO,CAAC,KAAK,EAAE,CAChB,CAAC;YACF,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAC5B,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,EAC/C,OAAO,CAAC,KAAK,EAAE,CAChB,CAAC;YACF,wBAAwB;YACxB,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAC;YACrB,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC;YACvB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC;YAC5B,OAAO,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC9B,CAAC;QAED,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC5B,CAAC;CACF"}

package/dist/mlx/cli.d.ts

@@ -0,0 +1,12 @@
+#!/usr/bin/env node
+/**
+ * parakeet.ts CLI
+ *
+ * Usage:
+ *   parakeet file.wav
+ *   parakeet file.wav --json
+ *   parakeet file.wav --model mlx-community/parakeet-tdt-0.6b-v3
+ *   parakeet --stream < pcm_f32le_16k.raw
+ */
+export {};
+//# sourceMappingURL=cli.d.ts.map

package/dist/mlx/cli.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.d.ts","sourceRoot":"","sources":["../../src/mlx/cli.ts"],"names":[],"mappings":";AACA;;;;;;;;GAQG"}

package/dist/mlx/cli.js

@@ -0,0 +1,145 @@
+#!/usr/bin/env node
+/**
+ * parakeet.ts CLI
+ *
+ * Usage:
+ *   parakeet file.wav
+ *   parakeet file.wav --json
+ *   parakeet file.wav --model mlx-community/parakeet-tdt-0.6b-v3
+ *   parakeet --stream < pcm_f32le_16k.raw
+ */
+import { parseArgs } from 'node:util';
+import { fromPretrained } from './load.js';
+import { consumePcmStream } from '../model.js';
+const DEFAULT_MODEL = 'mlx-community/parakeet-tdt-0.6b-v3';
+// ---------------------------------------------------------------------------
+// Arg parsing
+// ---------------------------------------------------------------------------
+const { values, positionals } = parseArgs({
+    options: {
+        model: { type: 'string', short: 'm', default: DEFAULT_MODEL },
+        json: { type: 'boolean', short: 'j', default: false },
+        stream: { type: 'boolean', short: 's', default: false },
+        help: { type: 'boolean', short: 'h', default: false },
+    },
+    allowPositionals: true,
+    args: process.argv.slice(2),
+});
+if (values.help) {
+    process.stdout.write(`
+parakeet — Nvidia Parakeet ASR (parakeet.ts)
+
+Usage:
+  parakeet <file.wav> [options]
+  parakeet --stream [options] < pcm_f32le_16k.raw
+
+Options:
+  --model, -m <id>   HuggingFace repo or local dir (default: ${DEFAULT_MODEL})
+  --json, -j         Output full AlignedResult JSON instead of plain text
+  --stream, -s       Read raw f32le mono 16kHz PCM from stdin
+  --help, -h         Show this help
+
+Exit codes: 0 success  1 file/IO error  2 model error
+`.trimStart());
+    process.exit(0);
+}
+// ---------------------------------------------------------------------------
+// Progress bar helpers
+// ---------------------------------------------------------------------------
+function progressBar(label, downloaded, total) {
+    if (!process.stderr.isTTY) {
+        return; // non-TTY: only emit the one-shot line (handled at call site)
+    }
+    if (total <= 0) {
+        process.stderr.write(`\r${label}: ${formatBytes(downloaded)}`);
+        return;
+    }
+    const pct = Math.min(100, Math.round((downloaded / total) * 100));
+    const barWidth = 30;
+    const filled = Math.round((pct / 100) * barWidth);
+    const bar = '█'.repeat(filled) + '░'.repeat(barWidth - filled);
+    process.stderr.write(`\r${label}: [${bar}] ${pct}% (${formatBytes(downloaded)}/${formatBytes(total)})`);
+}
+function formatBytes(n) {
+    if (n < 1024)
+        return `${n} B`;
+    if (n < 1024 * 1024)
+        return `${(n / 1024).toFixed(1)} KB`;
+    return `${(n / 1024 / 1024).toFixed(1)} MB`;
+}
+// ---------------------------------------------------------------------------
+// Main
+// ---------------------------------------------------------------------------
+async function main() {
+    const modelId = values.model ?? DEFAULT_MODEL;
+    // Load model
+    process.stderr.write(`Loading model: ${modelId}\n`);
+    let lastFile = '';
+    const model = await fromPretrained(modelId, {
+        onProgress(file, downloaded, total) {
+            if (file !== lastFile) {
+                if (lastFile !== '')
+                    process.stderr.write('\n'); // end previous file's line
+                if (!process.stderr.isTTY) {
+                    process.stderr.write(`Downloading ${file}...\n`);
+                }
+                lastFile = file;
+            }
+            progressBar(file, downloaded, total);
+            if (downloaded >= total && total > 0) {
+                if (process.stderr.isTTY)
+                    process.stderr.write('\n');
+            }
+        },
+    });
+    if (lastFile !== '' && process.stderr.isTTY) {
+        // Ensure cursor is on a new line after progress bars
+        process.stderr.write('\n');
+    }
+    process.stderr.write('Model ready.\n');
+    let result;
+    if (values.stream) {
+        // Streaming mode: read raw f32le PCM from stdin
+        process.stderr.write('Streaming PCM from stdin (16kHz mono f32le)...\n');
+        const stream = model.transcribeStream();
+        async function* stdinPcm() {
+            for await (const chunk of process.stdin) {
+                // Reinterpret raw bytes as Float32 (little-endian f32)
+                const buf = chunk instanceof Buffer ? chunk : Buffer.from(chunk);
+                // Ensure alignment: process only whole float32 values
+                const floatCount = Math.floor(buf.byteLength / 4);
+                if (floatCount > 0) {
+                    yield new Float32Array(buf.buffer, buf.byteOffset, floatCount);
+                }
+            }
+        }
+        result = await consumePcmStream(stream, stdinPcm());
+    }
+    else {
+        // File mode
+        const filePath = positionals[0];
+        if (!filePath) {
+            process.stderr.write('Error: provide a WAV file path or use --stream\n');
+            process.exit(1);
+        }
+        process.stderr.write(`Transcribing: ${filePath}\n`);
+        result = await model.transcribe(filePath);
+    }
+    // Output
+    if (values.json) {
+        process.stdout.write(JSON.stringify(result, null, 2) + '\n');
+    }
+    else {
+        process.stdout.write(result.text + '\n');
+    }
+}
+main().catch(err => {
+    const msg = err instanceof Error ? err.message : String(err);
+    if (msg.includes('ENOENT') || msg.includes('EACCES') || msg.includes('Failed to download')) {
+        process.stderr.write(`Error: ${msg}\n`);
+        process.exit(1);
+    }
+    process.stderr.write(`Model error: ${msg}\n`);
+    process.exit(2);
+});
+//# sourceMappingURL=cli.js.map

package/dist/mlx/cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../../src/mlx/cli.ts"],"names":[],"mappings":";AACA;;;;;;;;GAQG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAG/C,MAAM,aAAa,GAAG,oCAAoC,CAAC;AAE3D,8EAA8E;AAC9E,cAAc;AACd,8EAA8E;AAE9E,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,SAAS,CAAC;IACxC,OAAO,EAAE;QACP,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,aAAa,EAAE;QAC7D,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE;QACrD,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE;QACvD,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE;KACtD;IACD,gBAAgB,EAAE,IAAI;IACtB,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;CAC5B,CAAC,CAAC;AAEH,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;IAChB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;;;;;;;;+DAQwC,aAAa;;;;;;CAM3E,CAAC,SAAS,EAAE,CAAC,CAAC;IACb,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC;AAED,8EAA8E;AAC9E,uBAAuB;AACvB,8EAA8E;AAE9E,SAAS,WAAW,CAAC,KAAa,EAAE,UAAkB,EAAE,KAAa;IACnE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAC1B,OAAO,CAAC,8DAA8D;IACxE,CAAC;IACD,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACf,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,KAAK,KAAK,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAC/D,OAAO;IACT,CAAC;IACD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,UAAU,GAAG,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;IAClE,MAAM,QAAQ,GAAG,EAAE,CAAC;IACpB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;IAClD,MAAM,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,CAAC;IAC/D,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,KAAK,MAAM,GAAG,KAAK,GAAG,MAAM,WAAW,CAAC,UAAU,CAAC,IAAI,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;AAC1G,CAAC;AAED,SAAS,WAAW,CAAC,CAAS;IAC5B,IAAI,CAAC,GAAG,IAAI;QAAE,OAAO,GAAG,CAAC,IAAI,CAAC;IAC9B,IAAI,CAAC,GAAG,IAAI,GAAG,IAAI;QAAE,OAAO,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC;IAC1D,OAAO,GAAG,CAAC,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC;AAC9C,CAAC;AAED,8EAA8E;AAC9E,OAAO;AACP,8EAA8E;AAE9E,KAAK,UAAU,IAAI;IACjB,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,IAAI,aAAa,CAAC;IAE9C,aAAa;IACb,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,kBAAkB,OAAO,IAAI,CAAC,CAAC;IAEpD,IAAI,QAAQ,GAAG,EAAE,CAAC;IAClB,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE;QAC1C,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK;YAChC,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtB,IAAI,QAAQ,KAAK,EAAE;oBAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,2BAA2B;gBAC5E,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;oBAC1B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,IAAI,OAAO,CAAC,CAAC;gBACnD,CAAC;gBACD,QAAQ,GAAG,IAAI,CAAC;YAClB,CAAC;YACD,WAAW,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;YACrC,IAAI,UAAU,IAAI,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;gBACrC,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK;oBAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;KACF,CAAC,CAAC;IAEH,IAAI,QAAQ,KAAK,EAAE,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAC5C,qDAAqD;QACrD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;IAEvC,IAAI,MAAqB,CAAC;IAE1B,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QAClB,gDAAgD;QAChD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;QAEzE,MAAM,MAAM,GAAG,KAAK,CAAC,gBAAgB,EAAE,CAAC;QAExC,KAAK,SAAS,CAAC,CAAC,QAAQ;YACtB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,OAAO,CAAC,KAA8B,EAAE,CAAC;gBACjE,uDAAuD;gBACvD,MAAM,GAAG,GAAG,KAAK,YAAY,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACjE,sDAAsD;gBACtD,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;gBAClD,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;oBACnB,MAAM,IAAI,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,GAAG,MAAM,gBAAgB,CAAC,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IAEtD,CAAC;SAAM,CAAC;QACN,YAAY;QACZ,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;QAChC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;YACzE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,iBAAiB,QAAQ,IAAI,CAAC,CAAC;QACpD,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IAC5C,CAAC;IAED,SAAS;IACT,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QAChB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IAC/D,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;IAC3C,CAAC;AACH,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE;IACjB,MAAM,GAAG,GAAW,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACrE,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAC3F,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC;QACxC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IACD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAAC;IAC9C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/dist/mlx/conformer.d.ts

@@ -0,0 +1,81 @@
+import { MxArray } from '@mlx-node/core';
+import { Module, WeightMap, Linear, LayerNorm, BatchNorm, Conv1d, Conv2d } from './nn.js';
+import { MultiHeadAttention, RelPositionMultiHeadAttention, RelPositionMultiHeadLocalAttention, RelPositionalEncoding, LocalRelPositionalEncoding } from './attention.js';
+import { ConformerCache } from './cache.js';
+export interface ConformerArgs {
+    featIn: number;
+    nLayers: number;
+    dModel: number;
+    nHeads: number;
+    ffExpansionFactor: number;
+    subsamplingFactor: number;
+    selfAttentionModel: string;
+    subsampling: string;
+    convKernelSize: number;
+    subsamplingConvChannels: number;
+    posEmbMaxLen: number;
+    causalDownsampling?: boolean;
+    useBias?: boolean;
+    xscaling?: boolean;
+    subsamplingConvChunkingFactor?: number;
+    attContextSize?: [number, number] | null;
+}
+declare class FeedForward extends Module {
+    linear1: Linear;
+    linear2: Linear;
+    constructor(dModel: number, dFf: number, useBias: boolean);
+    forward(x: MxArray): MxArray;
+    loadWeights(weights: WeightMap, prefix: string): void;
+}
+declare class Convolution extends Module {
+    readonly padding: number;
+    pointwiseConv1: Conv1d;
+    depthwiseConv: Conv1d;
+    batchNorm: BatchNorm;
+    pointwiseConv2: Conv1d;
+    constructor(args: ConformerArgs);
+    forward(x: MxArray, cache: ConformerCache | null): MxArray;
+    loadWeights(weights: WeightMap, prefix: string): void;
+}
+type AttentionModel = 'rel_pos' | 'rel_pos_local_attn' | 'normal';
+export declare class ConformerBlock extends Module {
+    normFF1: LayerNorm;
+    ff1: FeedForward;
+    normSelfAtt: LayerNorm;
+    selfAttn: MultiHeadAttention | RelPositionMultiHeadAttention | RelPositionMultiHeadLocalAttention;
+    normConv: LayerNorm;
+    conv: Convolution;
+    normFF2: LayerNorm;
+    ff2: FeedForward;
+    normOut: LayerNorm;
+    private readonly args;
+    constructor(args: ConformerArgs);
+    private buildAttention;
+    setAttentionModel(name: AttentionModel, contextSize?: [number, number]): void;
+    forward(x: MxArray, posEmb: MxArray | null, mask: MxArray | null, cache: ConformerCache | null): MxArray;
+    loadWeights(weights: WeightMap, prefix: string): void;
+}
+declare class DwStridingSubsampling extends Module {
+    private readonly samplingNum;
+    private readonly stride;
+    private readonly kernelSize;
+    private readonly padding;
+    convLayers: Array<Conv2d | null>;
+    out: Linear;
+    constructor(args: ConformerArgs);
+    forward(x: MxArray, lengths: MxArray): [MxArray, MxArray];
+    loadWeights(weights: WeightMap, prefix: string): void;
+}
+export declare class Conformer extends Module {
+    readonly args: ConformerArgs;
+    posEnc: RelPositionalEncoding | LocalRelPositionalEncoding | null;
+    preEncode: DwStridingSubsampling | Linear;
+    layers: ConformerBlock[];
+    constructor(args: ConformerArgs);
+    setAttentionModel(name: AttentionModel, contextSize?: [number, number]): void;
+    forward(x: MxArray, // [batch, seq, mel]
+    lengths: MxArray | null, cache: Array<ConformerCache | null> | null): [MxArray, MxArray];
+    loadWeights(weights: WeightMap, prefix: string): void;
+}
+export {};
+//# sourceMappingURL=conformer.d.ts.map

package/dist/mlx/conformer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"conformer.d.ts","sourceRoot":"","sources":["../../src/mlx/conformer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EACL,MAAM,EACN,SAAS,EACT,MAAM,EACN,SAAS,EACT,SAAS,EACT,MAAM,EACN,MAAM,EAIP,MAAM,SAAS,CAAC;AACjB,OAAO,EACL,kBAAkB,EAClB,6BAA6B,EAC7B,kCAAkC,EAClC,qBAAqB,EACrB,0BAA0B,EAC3B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAU5C,MAAM,WAAW,aAAa;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,iBAAiB,EAAE,MAAM,CAAC;IAC1B,iBAAiB,EAAE,MAAM,CAAC;IAC1B,kBAAkB,EAAE,MAAM,CAAC;IAC3B,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,MAAM,CAAC;IACvB,uBAAuB,EAAE,MAAM,CAAC;IAChC,YAAY,EAAE,MAAM,CAAC;IACrB,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,6BAA6B,CAAC,EAAE,MAAM,CAAC;IACvC,cAAc,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC;CAC1C;AAMD,cAAM,WAAY,SAAQ,MAAM;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;gBAEJ,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO;IAMzD,OAAO,CAAC,CAAC,EAAE,OAAO,GAAG,OAAO;IAI5B,WAAW,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;CAItD;AAMD,cAAM,WAAY,SAAQ,MAAM;IAC9B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,SAAS,CAAC;IACrB,cAAc,EAAE,MAAM,CAAC;gBAEX,IAAI,EAAE,aAAa;IAW/B,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,cAAc,GAAG,IAAI,GAAG,OAAO;IAiB1D,WAAW,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;CAMtD;AAMD,KAAK,cAAc,GAAG,SAAS,GAAG,oBAAoB,GAAG,QAAQ,CAAC;AAElE,qBAAa,cAAe,SAAQ,MAAM;IACxC,OAAO,EAAE,SAAS,CAAC;IACnB,GAAG,EAAE,WAAW,CAAC;IAEjB,WAAW,EAAE,SAAS,CAAC;IACvB,QAAQ,EAAE,kBAAkB,GAAG,6BAA6B,GAAG,kCAAkC,CAAC;IAElG,QAAQ,EAAE,SAAS,CAAC;IACpB,IAAI,EAAE,WAAW,CAAC;IAElB,OAAO,EAAE,SAAS,CAAC;IACnB,GAAG,EAAE,WAAW,CAAC;IAEjB,OAAO,EAAE,SAAS,CAAC;IAEnB,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAgB;gBAEzB,IAAI,EAAE,aAAa;IAqB/B,OAAO,CAAC,cAAc;IAmBtB,iBAAiB,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,GAAE,CAAC,MAAM,EAAE,MAAM,CAAc,GAAG,IAAI;IAOzF,OAAO,CACL,CAAC,EAAE,OAAO,EACV,MAAM,EAAE,OAAO,GAAG,IAAI,EACtB,IAAI,EAAE,OAAO,GAAG,IAAI,EACpB,KAAK,EAAE,cAAc,GAAG,IAAI,GAC3B,OAAO;IAiBV,WAAW,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;CAWtD;AAMD,cAAM,qBAAsB,SAAQ,MAAM;IACxC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAK;IAC5B,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAK;IAChC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IAEjC,UAAU,EAAE,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IACjC,GAAG,EAAE,MAAM,CAAC;gBAEA,IAAI,EAAE,aAAa;IAkC/B,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC;IAiDzD,WAAW,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;CAUtD;AAMD,qBAAa,SAAU,SAAQ,MAAM;IACnC,QAAQ,CAAC,IAAI,EAAE,aAAa,CAAC;IAC7B,MAAM,EAAE,qBAAqB,GAAG,0BAA0B,GAAG,IAAI,CAAC;IAClE,SAAS,EAAE,qBAAqB,GAAG,MAAM,CAAC;IAC1C,MAAM,EAAE,cAAc,EAAE,CAAC;gBAEb,IAAI,EAAE,aAAa;IAiC/B,iBAAiB,CACf,IAAI,EAAE,cAAc,EACpB,WAAW,GAAE,CAAC,MAAM,EAAE,MAAM,CAAc,GACzC,IAAI;IAmBP,OAAO,CACL,CAAC,EAAE,OAAO,EAAE,oBAAoB;IAChC,OAAO,EAAE,OAAO,GAAG,IAAI,EACvB,KAAK,EAAE,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,IAAI,GACzC,CAAC,OAAO,EAAE,OAAO,CAAC;IAkCrB,WAAW,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;CAatD"}