@genai-fi/nanogpt 0.10.3 → 0.11.0

package/dist/{RealDiv-KAPDe8zB.js → RealDiv-Ds-jvL09.js}

@@ -1,10 +1,10 @@
-import { aG as T, j as E, af as O, h as V, aS as B, Y as F, am as j, aT as G } from "./index-DOvlwCh-.js";
-import { r as $ } from "./Reshape-Zt6eb7yh.js";
-import { a as A, b as k, d as C, c as N, e as R } from "./axis_util-BaG7mf5A.js";
-import { t as K, m as U } from "./shared-DT1TkE6w.js";
-import { c as W } from "./backend_util-RCe-rHaj.js";
-import { f as y } from "./gpgpu_math-DAOmgtXR.js";
-import { g as _, b as L } from "./kernel_funcs_utils-CCzYdUZg.js";
+import { aG as T, ab as E, af as O, V, aS as B, Q as F, am as G, aT as K } from "./index-D0RBWjq8.js";
+import { r as $ } from "./Reshape-Ct266DEk.js";
+import { a as A, b as k, d as C, c as N, e as R } from "./axis_util-DofAuy0p.js";
+import { t as U, m as W } from "./shared-B7USJZgw.js";
+import { c as _ } from "./backend_util-C7NWHpv7.js";
+import { f as y } from "./gpgpu_math-CJcbnKPC.js";
+import { g as j, b as L } from "./kernel_funcs_utils-CSaumNDs.js";
 class w {
   constructor(s, e) {
     this.variableNames = ["x"];
@@ -181,10 +181,10 @@ class X {
     `;
   }
 }
-function Y(a) {
+function q(a) {
   const s = [];
   for (; s.length === 0 || s[s.length - 1].outSize !== 1; ) {
-    const e = s.length ? s[s.length - 1].outSize : a[1], t = W(e);
+    const e = s.length ? s[s.length - 1].outSize : a[1], t = _(e);
     s.push({
       inSize: e,
       windowSize: t,
@@ -194,7 +194,7 @@ function Y(a) {
   return s;
 }
 function P(a, s, e, t) {
-  const n = Y(a.shape);
+  const n = q(a.shape);
   let l = a;
   for (let r = 0; r < n.length; r++) {
     const { inSize: i, windowSize: c, outSize: o } = n[r];
@@ -203,14 +203,14 @@ function P(a, s, e, t) {
   }
   return l;
 }
-class q {
+class Q {
   constructor(s, e) {
     this.variableNames = ["A"];
     const t = new Array(s.length);
     for (let r = 0; r < t.length; r++)
       t[r] = s[e[r]];
     this.outputShape = t, this.rank = t.length;
-    const n = y(this.rank), l = H(e);
+    const n = y(this.rank), l = Y(e);
     this.userCode = `
     void main() {
       ${n} resRC = getOutputCoords();
@@ -219,7 +219,7 @@ class q {
     `;
   }
 }
-function H(a) {
+function Y(a) {
   const s = a.length;
   if (s > 6)
     throw Error(`Transpose for rank ${s} is not yet supported`);
@@ -228,7 +228,7 @@ function H(a) {
     t[a[n]] = e[n];
   return t.join();
 }
-class J {
+class H {
   constructor(s, e) {
     this.variableNames = ["A"], this.packedInputs = !0, this.packedOutput = !0;
     const t = new Array(s.length);
@@ -236,7 +236,7 @@ class J {
       t[u] = s[e[u]];
     if (this.outputShape = t, this.rank = t.length, this.rank > 6)
       throw Error(`Packed transpose for rank ${this.rank} is not yet supported.`);
-    const n = y(this.rank), l = _("rc", this.rank), r = new Array(this.rank);
+    const n = y(this.rank), l = j("rc", this.rank), r = new Array(this.rank);
     for (let u = 0; u < e.length; u++)
       r[e[u]] = l[u];
     const i = `vec2(${r.slice(-2).join()})`, c = `++${l[this.rank - 1]} < ${t[this.rank - 1]}`, o = `getChannel(getA(${r.join()}), ${i})`;
@@ -261,10 +261,10 @@ class J {
   }
 }
 function D(a, s, e) {
-  const t = E().getBool("WEBGL_PACK_ARRAY_OPERATIONS") ? new J(a.shape, s) : new q(a.shape, s);
+  const t = E().getBool("WEBGL_PACK_ARRAY_OPERATIONS") ? new H(a.shape, s) : new Q(a.shape, s);
   return e.runWebGLProgram(t, [a], a.dtype);
 }
-function Q(a, s, e, t) {
+function J(a, s, e, t) {
   const n = s, l = a.shape.length, r = O(n, a.shape);
   let i = r;
   const c = A(i, l), o = c != null;
@@ -278,7 +278,7 @@ function Q(a, s, e, t) {
 }
 function Z(a) {
   const { inputs: s, backend: e, attrs: t } = a, { x: n } = s, { axis: l, keepDims: r } = t;
-  return Q(n, l, r, e);
+  return J(n, l, r, e);
 }
 const pe = {
   kernelName: F,
@@ -299,7 +299,7 @@ function te(a) {
       const I = e.texData.get(d.dataId).values, m = new Array(i);
       for (let v = 0; v < m.length; v++)
         m[v] = n.shape[u[v]];
-      const z = K(I, n.shape, n.dtype, u, m);
+      const z = U(I, n.shape, n.dtype, u, m);
       d = e.makeTensorInfo(m, n.dtype);
       const M = e.texData.get(d.dataId);
       M.values = z;
@@ -313,7 +313,7 @@ function te(a) {
   r && (g = R(f, c));
   let x;
   if (h) {
-    const I = e.texData.get(d.dataId).values, m = U(I, V(S), g, n.dtype);
+    const I = e.texData.get(d.dataId).values, m = W(I, V(S), g, n.dtype);
     x = e.makeTensorInfo(g, n.dtype);
     const z = e.texData.get(x.dataId);
     z.values = m;
@@ -322,7 +322,7 @@ function te(a) {
   return p && e.disposeIntermediateTensorInfo(d), x;
 }
 const he = {
-  kernelName: j,
+  kernelName: G,
   backendName: "webgl",
   kernelFunc: te
 };
@@ -349,7 +349,7 @@ return a / b;`, se = `
 
   return result;
 `, ne = L({ opSnippet: ae, packedOpSnippet: se, checkOutOfBounds: !0 }), fe = {
-  kernelName: G,
+  kernelName: K,
   backendName: "webgl",
   kernelFunc: ne
 };

package/dist/{Reshape-BYkmUnAv.js → Reshape-Cd6e-Otn.js}

@@ -1,4 +1,4 @@
-import { h, ah as d, l as c, R as m } from "./index-DOvlwCh-.js";
+import { V as h, ah as d, y as c, R as m } from "./index-D0RBWjq8.js";
 function i(n) {
   const { inputs: p, attrs: o } = n, { x: e } = p, { shape: r } = o, a = h(e.shape), s = d(r, a), t = h(s);
   return c(a === t, () => `The new shape (${s}) has ${t} elements and the old shape (${e.shape}) has ${a} elements. The new shape and old shape must have the same number of elements.`), n.backend.incRef(e.dataId), { dataId: e.dataId, shape: s, dtype: e.dtype };

package/dist/{Reshape-Zt6eb7yh.js → Reshape-Ct266DEk.js}

@@ -1,5 +1,5 @@
-import { R as C, h as c, ah as R, l as f } from "./index-DOvlwCh-.js";
-import { u as g, g as I, a as x, b as F, c as $, d as u, e as l, i as m } from "./gpgpu_math-DAOmgtXR.js";
+import { R as C, V as c, ah as R, y as f } from "./index-D0RBWjq8.js";
+import { u as g, g as I, a as x, b as F, c as $, d as u, e as m, i as l } from "./gpgpu_math-CJcbnKPC.js";
 class S {
   constructor(t, i) {
     this.variableNames = ["A"], this.packedInputs = !0, this.packedOutput = !0, this.customUniforms = [{ name: "inputShape", type: "ivec3" }], this.outputShape = t, this.enableShapeUniforms = g(this.outputShape.length);
@@ -47,33 +47,33 @@ function v(s, t) {
     }
   `;
 }
-function b(s, t, i) {
+function y(s, t, i) {
   const a = [
     u(s.shape),
-    ...l(s.shape)
+    ...m(s.shape)
   ], e = {
     dtype: s.dtype,
     shape: a,
     dataId: s.dataId
   }, o = [
     u(t),
-    ...l(t)
+    ...m(t)
   ], r = new S(o, a), p = !0, n = [a], h = i.runWebGLProgram(r, [e], s.dtype, n, p);
   return { dataId: h.dataId, shape: t, dtype: h.dtype };
 }
-function y(s) {
+function b(s) {
   const { inputs: t, backend: i, attrs: a } = s, { x: e } = t, { shape: o } = a, r = i, p = c(e.shape), n = R(o, p), h = c(n);
   f(p === h, () => `The new shape (${n}) has ${h} elements and the old shape (${e.shape}) has ${p} elements. The new shape and old shape must have the same number of elements.`);
   const d = r.texData.get(e.dataId);
-  return d.isPacked && !m(e.shape, n) && !(d.texture !== null && m(d.shape, n)) ? b(e, n, r) : (r.incRef(e.dataId), { dataId: e.dataId, shape: n, dtype: e.dtype });
+  return d.isPacked && !l(e.shape, n) && !(d.texture !== null && l(d.shape, n)) ? y(e, n, r) : (r.incRef(e.dataId), { dataId: e.dataId, shape: n, dtype: e.dtype });
 }
 const U = {
   kernelName: C,
   backendName: "webgl",
-  kernelFunc: y
+  kernelFunc: b
 };
 export {
   S as R,
   U as a,
-  y as r
+  b as r
 };

package/dist/TeachableLLM.d.ts

@@ -1,5 +1,5 @@
 import { GPTConfig } from './models/config';
-import { ITokeniser } from './tokeniser/type';
+import { Conversation, ITokeniser } from './tokeniser/type';
 import { SaveOptions } from './loader/save';
 import { default as Generator, IGenerateOptions } from './Generator';
 import { default as Trainer, ITrainerOptions } from './Trainer';
@@ -41,10 +41,11 @@ export default class TeachableLLM {
     set enableProfiler(value: boolean);
     getNumParams(): number;
     trainer(): Trainer;
-    train(text: string[], options?: ITrainerOptions): Promise<void>;
+    train(text: Conversation[][], options?: ITrainerOptions): Promise<void>;
     trainTokeniser(text: string[]): Promise<number>;
     generator(): Generator;
-    generateText(prompt?: string, options?: IGenerateOptions): Promise<string>;
+    generateText(prompt: Conversation[], options?: IGenerateOptions): Promise<Conversation[]>;
+    generateText(options?: IGenerateOptions): Promise<Conversation[]>;
     dispose(): void;
     on(event: 'status', listener: (status: TeachableLLMStatus) => void): void;
     on(event: 'error', listener: (error: Error) => void): void;

package/dist/TeachableLLM.js

@@ -2,27 +2,27 @@ import { defaultConfig as d } from "./models/config.js";
 import { saveModel as l } from "./loader/save.js";
 import { loadModel as _ } from "./loader/load.js";
 import u from "./Generator.js";
-import p from "./Trainer.js";
-import { E as f } from "./index-DvYrXKkX.js";
+import f from "./Trainer.js";
+import { E as p } from "./index-DvYrXKkX.js";
 import { dummyPassTrainAsync as m } from "./utilities/dummy.js";
-import "./index-DOvlwCh-.js";
-import "./random_width-or-CEftb.js";
-import "./zeros_like-DvE73F4e.js";
+import "./index-D0RBWjq8.js";
+import "./random_width-UGQn4OWb.js";
+import "./zeros_like-BAz3iKru.js";
 import "./index-Cp39cXWe.js";
-import "./dataset-BcwmTGYc.js";
+import "./dataset-DcjWqUVQ.js";
 import "./ops/cpu/attentionMask.js";
 import "./ops/webgl/attentionMask.js";
 import "./ops/grads/attentionMask.js";
 import "./ops/cpu/rope.js";
 import "./ops/webgl/rope.js";
-import "./rope-Ir4mTyD1.js";
+import "./rope-BmZmp9uP.js";
 import "./ops/cpu/appendCache.js";
 import "./ops/webgl/appendCache.js";
 import "./ops/grads/softmax16.js";
-import "./matMul16-BWRSOCWB.js";
+import "./matMul16-cDxwemKj.js";
 import "./ops/webgl/matMul16.js";
 import "./ops/cpu/matMul16.js";
-import "./pack16-nQ6JaLo-.js";
+import "./pack16-k4jq6aMX.js";
 import "./ops/transpose16.js";
 import "./ops/reshape16.js";
 import "./ops/cpu/qkv.js";
@@ -41,11 +41,11 @@ import g from "./tokeniser/bpe.js";
 import "./papaparse.min-C0cScC2i.js";
 import "./jszip.min-Bz5-11Bk.js";
 import "./ops/cpu/matMulGelu.js";
-import "./matMulGelu-CzfgT6Wq.js";
+import "./matMulGelu-B2s_80-H.js";
 import "./ops/grads/matMulGelu.js";
 import "./ops/cpu/gelu.js";
 import "./ops/webgl/gelu.js";
-import "./gelu-CjNPL4OH.js";
+import "./gelu-DqTbCx5x.js";
 import "./ops/webgl/log.js";
 import "./ops/cpu/adamMoments.js";
 import "./ops/webgl/adamMoments.js";
@@ -56,7 +56,7 @@ import "./checks/normRMSGrad.js";
 import k from "./utilities/profile.js";
 import w from "./models/factory.js";
 class a {
-  ee = new f();
+  ee = new p();
   _config;
   _model;
   _tokeniser;
@@ -155,7 +155,7 @@ class a {
   trainer() {
     if (!this._model || !this._tokeniser)
       throw new Error("model_or_tokeniser_not_initialized.");
-    const t = new p(this._model, this._tokeniser);
+    const t = new f(this._model, this._tokeniser);
     return t.on("start", () => this.setStatus("training")), t.on("stop", () => this.setStatus("ready")), t.on("log", async (e, r) => {
       const o = this.ee.listeners("trainStep");
       for (const s of o)
@@ -183,7 +183,7 @@ class a {
     }), t;
   }
   generateText(t, e) {
-    return this.generator().generate(t, e);
+    return Array.isArray(t) ? this.generator().generate(t, e) : this.generator().generate([], e);
   }
   dispose() {
     this._model?.dispose();

package/dist/Trainer.d.ts

@@ -1,4 +1,4 @@
-import { ITokeniser } from './tokeniser/type';
+import { Conversation, ITokeniser } from './tokeniser/type';
 import { default as EE } from 'eventemitter3';
 import { TrainingLogEntry, TrainingProgress } from './training/Trainer';
 import { default as Model, ModelForwardAttributes } from './models/model';
@@ -30,7 +30,7 @@ export default class Trainer extends EE<'start' | 'stop' | 'log'> {
     constructor(model: Model<ModelForwardAttributes>, tokeniser: ITokeniser);
     stop(): void;
     reset(): void;
-    prepare(text: string[], options?: ITrainerOptions): Promise<void>;
+    prepare(text: Conversation[][], options?: ITrainerOptions): Promise<void>;
     train(options?: ITrainerOptions): Promise<void>;
     step(options?: ITrainerOptions): Promise<void>;
     getLog(): TrainingLogEntry[];

package/dist/Trainer.js

@@ -1,6 +1,6 @@
-import { E as l } from "./index-DvYrXKkX.js";
-import h from "./training/FullTrainer.js";
-class m extends l {
+import { E as o } from "./index-DvYrXKkX.js";
+import d from "./training/FullTrainer.js";
+class g extends o {
   trainer;
   hasTrained = !1;
   trainDataset;
@@ -9,7 +9,7 @@ class m extends l {
   log = [];
   progress = null;
   constructor(t, e) {
-    super(), this.trainer = new h(t, e, 1e-3);
+    super(), this.trainer = new d(t, e, 1e-3);
   }
   stop() {
     this.trainer.stop();
@@ -22,7 +22,7 @@ class m extends l {
       t,
       e?.batchSize || 32,
       e?.validationSplit || 0.1
-    ), i = t.reduce((r, n) => r + n.length, 0) * (1 - (e?.validationSplit || 0));
+    ), i = t.reduce((r, n) => r + n.reduce((l, h) => l + h.content.length, 0), 0) * (1 - (e?.validationSplit || 0));
     this.trainDataset = a, this.validationDataset = s, this.totalSamples = i;
   }
   async train(t) {
@@ -91,5 +91,5 @@ class m extends l {
   }
 }
 export {
-  m as default
+  g as default
 };

package/dist/{axis_util-BaG7mf5A.js → axis_util-DofAuy0p.js}

@@ -1,4 +1,4 @@
-import { l as c } from "./index-DOvlwCh-.js";
+import { y as c } from "./index-D0RBWjq8.js";
 function i(e, n) {
   for (let t = 0; t < e.length; ++t)
     if (e[e.length - t - 1] !== n - 1 - t)
@@ -12,7 +12,7 @@ function a(e, n, t) {
     t.indexOf(u) === -1 ? s.push(e[o++]) : s.push(n[f++]);
   return s;
 }
-function p(e, n) {
+function l(e, n) {
   const t = [], r = e.length;
   for (let o = 0; o < r; o++)
     n.indexOf(o) === -1 && t.push(e[o]);
@@ -46,7 +46,7 @@ function x(e, n) {
 export {
   d as a,
   x as b,
-  p as c,
+  l as c,
   m as d,
   h as e,
   i as f,

package/dist/backend.js

@@ -1,9 +1,9 @@
-import { g as o, s as e, r as s } from "./index-DOvlwCh-.js";
+import { g as o, s as e, r as s } from "./index-D0RBWjq8.js";
 async function c(t, a) {
   if (o() !== t) {
     if (t === "webgpu") {
       const { registerWebGPUBackend: i } = await import("./patches/webgpu_base.js");
-      i(a), await import("./index-BwexR4lA.js"), await import("./ops/webgpu/index.js");
+      i(a), await import("./index-Dj5TkmPY.js"), await import("./ops/webgpu/index.js");
     }
     await e(t), await s(), console.log(`Backend set to ${t}`);
   }

package/dist/{backend_util-RCe-rHaj.js → backend_util-C7NWHpv7.js}

@@ -1,10 +1,10 @@
-import { h as m, ad as w, aU as I, l as d, ax as A, aB as _, a3 as y, o as M, a4 as T, aV as b, ak as D, aW as x } from "./index-DOvlwCh-.js";
-import { d as L, f as W, h as v, c as F, e as N, a as C, b as P, g as z } from "./axis_util-BaG7mf5A.js";
-import { a as B, c as U } from "./concat_util-DpW8mL_l.js";
-import { c as V, b as G, d as H, f as j, g as q, h as Z, i as k, j as J, k as K, m as X, t as Y } from "./step-CINUs5QB.js";
-import { S as Q, a as ee, b as te, g as se, c as ne, s as re } from "./selu_util-kbhpTdYD.js";
-import { s as oe } from "./slice_util-uTKwiEpW.js";
-import { c as ae, v as ie, a as ue } from "./scatter_nd_util-lvSiX8q4.js";
+import { V as m, a9 as w, aU as I, y as d, ax as A, aB as _, $ as y, ad as M, a0 as T, aV as b, ak as D, aW as x } from "./index-D0RBWjq8.js";
+import { d as L, f as W, h as v, c as F, e as N, a as C, b as P, g as z } from "./axis_util-DofAuy0p.js";
+import { a as B, c as U } from "./concat_util-CHsJFZJJ.js";
+import { c as V, b as G, d as H, f as j, g as q, h as Z, i as k, j as J, k as K, m as X, t as Y } from "./step-dXR33iOg.js";
+import { S as Q, a as ee, b as te, g as se, c as ne, s as re } from "./selu_util-BuLbmbrl.js";
+import { s as oe } from "./slice_util-D8CQRenR.js";
+import { c as ae, v as ie, a as ue } from "./scatter_nd_util-BY22Cc-C.js";
 import { a as le, c as pe, b as ce, e as he, d as fe, g as ge, m as de, s as me } from "./complex_util-Yc1A_gV1.js";
 function Ee(e, t) {
   const r = e.shape.length, s = t.shape.length;

package/dist/{backend_webgpu-DE3ACOLx.js → backend_webgpu-B0Vls736.js}

@@ -1,6 +1,6 @@
-import { j as g, au as $, av as K, e as D, l as _, aw as O, h as x, ax as Z, at as W, ay as F, az as j, aA as X, aB as J, i as ee, ad as k } from "./index-DOvlwCh-.js";
-import { m as te, f as se, P as re } from "./webgpu_program-DuOXPQol.js";
-import { i as ne, G as q } from "./webgpu_util-RxEF33Rj.js";
+import { ab as g, au as $, av as K, e as D, y as _, aw as O, V as x, ax as Z, at as W, ay as F, az as j, aA as X, aB as J, ae as ee, a9 as k } from "./index-D0RBWjq8.js";
+import { m as te, f as se, P as re } from "./webgpu_program-CAE4RICo.js";
+import { i as ne, G as q } from "./webgpu_util-BdovYhXr.js";
 import { m as N } from "./complex_util-Yc1A_gV1.js";
 const d = g();
 d.registerFlag("WEBGPU_DEFERRED_SUBMIT_BATCH_SIZE", () => 15);
@@ -264,7 +264,7 @@ class R extends $ {
         alphaMode: r[B]
       }), y.getCurrentTexture();
     }).map((E, B) => {
-      const y = f * 4, b = (P, S, v) => {
+      const y = f * 4, G = (P, S, v) => {
         this.ensureCommandEncoderReady(), this.commandEncoder.copyBufferToTexture({
           buffer: a,
           bytesPerRow: y,
@@ -279,20 +279,20 @@ class R extends $ {
           willReadFrequently: !0
         });
         I.clearRect(0, 0, P, S), I.drawImage(h[B], 0, 0);
-        const G = I.getImageData(0, 0, P, S).data, H = r[B], M = new Uint8ClampedArray(o, v, P * S * 4);
+        const b = I.getImageData(0, 0, P, S).data, H = r[B], M = new Uint8ClampedArray(o, v, P * S * 4);
         for (let p = 0; p < M.length; p += 4)
           if (H === "premultiplied")
-            M[p + 3] = G[p + 3];
+            M[p + 3] = b[p + 3];
           else {
-            const V = G[p];
-            M[p] = G[p + 2], M[p + 1] = G[p + 1], M[p + 2] = V;
+            const V = b[p];
+            M[p] = b[p + 2], M[p + 1] = b[p + 1], M[p + 2] = V;
           }
       }, Y = Math.floor(u / (f * c));
       let T = f, U = c, C = 0;
       for (let P = 0; P < Y; P++)
-        b(T, U, C), C += f * c * 4;
+        G(T, U, C), C += f * c * 4;
       const A = u % (f * c);
-      U = Math.floor(A / f), U > 0 && (b(T, U, C), C += U * (f * 4)), T = A % f, T > 0 && b(T, 1, C);
+      U = Math.floor(A / f), U > 0 && (G(T, U, C), C += U * (f * 4)), T = A % f, T > 0 && G(T, 1, C);
     });
     const w = O(o, t.dtype);
     return this.convertAndCacheOnCPU(e, w), w;

package/dist/{broadcast_to-B3eYlZm7.js → broadcast_to-DDaNMbX7.js}

@@ -1,5 +1,5 @@
-import { A as h, B as f, C as p, D as g, E as u, T } from "./index-DOvlwCh-.js";
-import { r as b } from "./reshape-ByE68wS9.js";
+import { q as h, u as f, w as p, x as g, E as u, T } from "./index-D0RBWjq8.js";
+import { r as b } from "./reshape-CkjKPPqB.js";
 function m(e, r) {
   let n = f(e, "broadcastTo", "x");
   const a = n.shape;

package/dist/checks/appendCache.js

@@ -1,5 +1,5 @@
-import { s, e as a } from "../index-DOvlwCh-.js";
-import { t } from "../tensor4d-BARPdTaS.js";
+import { s, e as a } from "../index-D0RBWjq8.js";
+import { t } from "../tensor4d-DVwr7pLF.js";
 async function u(e) {
   await s(e);
   const n = t(

package/dist/checks/attentionMask.js

@@ -1,6 +1,6 @@
-import { s as i, e } from "../index-DOvlwCh-.js";
-import { t } from "../tensor4d-BARPdTaS.js";
-import { t as a } from "../tensor2d-Bs9wZRc7.js";
+import { s as i, e } from "../index-D0RBWjq8.js";
+import { t } from "../tensor4d-DVwr7pLF.js";
+import { t as a } from "../tensor2d-BN1sSfQO.js";
 async function k(n) {
   await i(n);
   const s = t(

package/dist/checks/gelu.js

@@ -1,5 +1,5 @@
-import { s as e, e as o } from "../index-DOvlwCh-.js";
-import { t as s } from "../tensor2d-Bs9wZRc7.js";
+import { s as e, e as o } from "../index-D0RBWjq8.js";
+import { t as s } from "../tensor2d-BN1sSfQO.js";
 async function m(t) {
   await e(t);
   const r = s(

package/dist/checks/matMulGelu.js

@@ -1,5 +1,5 @@
-import { s as o, e as s } from "../index-DOvlwCh-.js";
-import { t as e } from "../tensor2d-Bs9wZRc7.js";
+import { s as o, e as s } from "../index-D0RBWjq8.js";
+import { t as e } from "../tensor2d-BN1sSfQO.js";
 async function i(t) {
   await o(t);
   const r = e(

package/dist/checks/normRMS.js

@@ -1,7 +1,7 @@
-import { s as u, a5 as A, e as y } from "../index-DOvlwCh-.js";
-import { a as h } from "../ops-B5yanEdW.js";
-import { t as p } from "../tensor1d-D11P_7Dp.js";
-import { t as r } from "../tensor-DJoc7gJU.js";
+import { s as u, a1 as A, e as y } from "../index-D0RBWjq8.js";
+import { a as h } from "../ops-FJapAPfm.js";
+import { t as p } from "../tensor1d-LxP9asMm.js";
+import { t as r } from "../tensor-BQqrDvpx.js";
 const w = Array.from({ length: 2048 * 192 }, () => Math.random()), x = Array.from({ length: 192 }, () => Math.random()), M = Array.from({ length: 2048 * 192 }, () => Math.random());
 async function k(t) {
   await u(t);

package/dist/checks/normRMSGrad.js

@@ -1,6 +1,6 @@
-import { s as c, e as d } from "../index-DOvlwCh-.js";
-import { t as f } from "../tensor1d-D11P_7Dp.js";
-import { t as r } from "../tensor-DJoc7gJU.js";
+import { s as c, e as d } from "../index-D0RBWjq8.js";
+import { t as f } from "../tensor1d-LxP9asMm.js";
+import { t as r } from "../tensor-BQqrDvpx.js";
 const y = Array.from({ length: 2048 * 192 }, () => Math.random()), i = Array.from({ length: 192 }, () => Math.random()), l = Array.from({ length: 2048 * 192 }, () => Math.random());
 async function x(t) {
   await c(t);

package/dist/checks/packUnpack.js

@@ -1,5 +1,5 @@
-import { s as a, e } from "../index-DOvlwCh-.js";
-import { t as c } from "../tensor2d-Bs9wZRc7.js";
+import { s as a, e } from "../index-D0RBWjq8.js";
+import { t as c } from "../tensor2d-BN1sSfQO.js";
 async function i(n) {
   await a(n);
   const r = c(

package/dist/checks/qkv.js

@@ -1,5 +1,5 @@
-import { $ as i, a0 as u, a1 as c, s as l, e as h } from "../index-DOvlwCh-.js";
-import { t as f } from "../tensor2d-Bs9wZRc7.js";
+import { X as i, Y as u, Z as c, s as l, e as h } from "../index-D0RBWjq8.js";
+import { t as f } from "../tensor2d-BN1sSfQO.js";
 function m(t, e, n) {
   if (i(t), e != null && e.length !== 3)
     throw new Error("tensor3d() requires shape to have three numbers");
@@ -26,8 +26,8 @@ async function y(t) {
       [0.7, 0.8, 1.1, 1.2, 1.5, 1.6]
     ],
     [2, 6]
-  ), r = h().runKernel("QKV", { x: e, kernel: n }, { heads: 1 }), a = await r[0].array(), o = await r[1].array(), s = await r[2].array();
-  return [a, o, s];
+  ), r = h().runKernel("QKV", { x: e, kernel: n }, { heads: 1 }), o = await r[0].array(), a = await r[1].array(), s = await r[2].array();
+  return [o, a, s];
 }
 export {
   y as execute

package/dist/checks/rope.js

@@ -1,6 +1,6 @@
 import t from "../layers/RoPECache.js";
-import { s as c, e as i } from "../index-DOvlwCh-.js";
-import { t as p } from "../tensor4d-BARPdTaS.js";
+import { s as c, e as i } from "../index-D0RBWjq8.js";
+import { t as p } from "../tensor4d-DVwr7pLF.js";
 async function y(a) {
   await c(a);
   const o = p(

package/dist/{clip_by_value-BnO7-a88.js → clip_by_value-Dn5tzexi.js}

@@ -1,8 +1,8 @@
-import { A as a, B as m, l as y, q as B, E as c, F as f } from "./index-DOvlwCh-.js";
+import { q as a, u as y, y as m, i as u, E as c, C as f } from "./index-D0RBWjq8.js";
 function p(o, s, t) {
-  const r = m(o, "x", "clipByValue");
-  if (y(s <= t, () => `Error in clip: min (${s}) must be less than or equal to max (${t}).`), s === t)
-    return B(r.shape, s, r.dtype);
+  const r = y(o, "x", "clipByValue");
+  if (m(s <= t, () => `Error in clip: min (${s}) must be less than or equal to max (${t}).`), s === t)
+    return u(r.shape, s, r.dtype);
   const n = { x: r }, e = { clipValueMin: s, clipValueMax: t };
   return c.runKernel(f, n, e);
 }

package/dist/complex-DClmWqJt.js

@@ -0,0 +1,11 @@
+import { q as c, u as s, a7 as n, E as m, a8 as r } from "./index-D0RBWjq8.js";
+function l(o, p) {
+  const a = s(o, "real", "complex"), e = s(p, "imag", "complex");
+  n(a.shape, e.shape, `real and imag shapes, ${a.shape} and ${e.shape}, must match in call to tf.complex().`);
+  const t = { real: a, imag: e };
+  return m.runKernel(r, t);
+}
+const i = /* @__PURE__ */ c({ complex_: l });
+export {
+  i as c
+};

package/dist/{concat-BV8bt5H-.js → concat-C6X3AAlQ.js}

@@ -1,4 +1,4 @@
-import { A as s, l as a, G as p, D as i, E as l, H as f } from "./index-DOvlwCh-.js";
+import { q as s, y as a, z as p, x as i, E as l, A as f } from "./index-D0RBWjq8.js";
 function h(n, e = 0) {
   a(n.length >= 1, () => "Pass at least one tensor to concat");
   const t = p(n, "tensors", "concat", "string_or_numeric");

package/dist/{concat_util-DpW8mL_l.js → concat_util-CHsJFZJJ.js}

@@ -1,4 +1,4 @@
-import { l as s } from "./index-DOvlwCh-.js";
+import { y as s } from "./index-D0RBWjq8.js";
 function h(n, o) {
   const t = n[0].length;
   n.forEach((a, c) => {

package/dist/{dataset-BcwmTGYc.js → dataset-DcjWqUVQ.js}

@@ -1,7 +1,7 @@
-import { aC as S, a8 as h, j as N, d as v, aD as o, aE as p, aF as g, l as k, t as y } from "./index-DOvlwCh-.js";
+import { aC as S, a4 as h, ab as N, d as v, aD as o, aE as p, aF as g, y as k, t as y } from "./index-D0RBWjq8.js";
 import { s as R } from "./index-CieiGp4Y.js";
-import { s as $ } from "./stack-Cf4n9h0N.js";
-import { t as B } from "./tensor-DJoc7gJU.js";
+import { s as $ } from "./stack-WJK22CFn.js";
+import { t as B } from "./tensor-BQqrDvpx.js";
 function _(s, t) {
   return b(s, t);
 }
@@ -296,7 +296,7 @@ class i {
    *   throw an `Error`.
    */
   handleErrors(t) {
-    return new j(this, t);
+    return new Z(this, t);
   }
   // TODO(soergel): Implement reduce() etc.
   /**
@@ -623,7 +623,7 @@ class K extends i {
     return { value: r, done: !1 };
   }
 }
-class j extends i {
+class Z extends i {
   constructor(t, e) {
     super(), this.upstream = t, this.handler = e, this.count = 0, this.lastRead = Promise.resolve({ value: null, done: !1 });
   }
@@ -660,7 +660,7 @@ class w extends i {
     return { value: r, done: !1 };
   }
 }
-class Z extends i {
+class j extends i {
   constructor() {
     super(), this.outputQueue = new f(), this.lastRead = Promise.resolve({ value: null, done: !1 });
   }
@@ -674,7 +674,7 @@ class Z extends i {
     return { value: this.outputQueue.shift(), done: !1 };
   }
 }
-class tt extends Z {
+class tt extends j {
   constructor(t, e) {
     super(), this.upstream = t, this.transform = e;
   }