@genai-fi/nanogpt 0.17.4 → 0.18.0

package/dist/{clip_by_value-DDA7rrcT.js → clip_by_value-CLAD4h_I.js}

@@ -1,4 +1,4 @@
-import { o as a, q as f, x as m, f as y, E as c, C as p } from "./index-CUXkjxiT.js";
+import { o as a, n as f, v as m, f as y, E as c, C as p } from "./index-DSGwv2Yx.js";
 function u(o, s, t) {
   const r = f(o, "x", "clipByValue");
   if (m(s <= t, () => `Error in clip: min (${s}) must be less than or equal to max (${t}).`), s === t)

package/dist/complex-3DpPEG9B.js

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

package/dist/{concat-CAQpCret.js → concat-Dqk7Xk7h.js}

@@ -1,5 +1,5 @@
-import { o as s, x as a, y as p, w as i, E as l, z as f } from "./index-CUXkjxiT.js";
-function h(o, e = 0) {
+import { o as s, v as a, w as p, u as i, E as l, x as u } from "./index-DSGwv2Yx.js";
+function f(o, e = 0) {
   a(o.length >= 1, () => "Pass at least one tensor to concat");
   const t = p(o, "tensors", "concat", "string_or_numeric");
   if (t[0].dtype === "complex64" && t.forEach((n) => {
@@ -9,9 +9,9 @@ function h(o, e = 0) {
   }), t.length === 1)
     return i(t[0]);
   const r = t, c = { axis: e };
-  return l.runKernel(f, r, c);
+  return l.runKernel(u, r, c);
 }
-const u = /* @__PURE__ */ s({ concat_: h });
+const m = /* @__PURE__ */ s({ concat_: f });
 export {
-  u as c
+  m as c
 };

package/dist/{concat_util-D18dJ4fD.js → concat_util-C1Mxe27t.js}

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

package/dist/{dataset-CGGp1z9P.js → dataset-DlqAN81i.js}

@@ -1,7 +1,7 @@
-import { ag as h, aA as S, ab as N, d as v, aB as o, aC as p, aD as g, x as k, t as y } from "./index-CUXkjxiT.js";
+import { ag as h, aA as S, ab as N, l as v, aB as o, aC as p, aD as g, v as k, t as y } from "./index-DSGwv2Yx.js";
 import { s as R } from "./index-CieiGp4Y.js";
-import { s as B } from "./stack-DudVrtmG.js";
-import { t as $ } from "./tensor-BWFldCso.js";
+import { s as B } from "./stack-z6QE7kmP.js";
+import { t as $ } from "./tensor-D8e0Gd7c.js";
 function _(s, t) {
   return b(s, t);
 }

package/dist/{dropout_util--NxWuYg2.js → dropout_util-N0z8Os-K.js}

@@ -1,4 +1,4 @@
-import { o as u, u as f, a5 as i, ac as m } from "./index-CUXkjxiT.js";
+import { o as u, q as f, a5 as i, ac as m } from "./index-DSGwv2Yx.js";
 import { U as c } from "./rand_util-CZ7yLoUm.js";
 function g(n, r = 0, t = 1, e = "float32", a) {
   f(n);

package/dist/{expand_dims-Bkd1YD5x.js → expand_dims-D0rBtgT1.js}

@@ -1,11 +1,11 @@
-import { o as e, q as a, x as i, E as m, A as p } from "./index-CUXkjxiT.js";
-function x(t, n = 0) {
+import { o as e, n as a, v as i, E as m, y as p } from "./index-DSGwv2Yx.js";
+function c(t, n = 0) {
   const s = a(t, "x", "expandDims", "string_or_numeric");
   i(n <= s.rank, () => "Axis must be <= rank of the tensor");
   const r = { input: s }, o = { dim: n };
   return m.runKernel(p, r, o);
 }
-const u = /* @__PURE__ */ e({ expandDims_: x });
+const x = /* @__PURE__ */ e({ expandDims_: c });
 export {
-  u as e
+  x as e
 };

package/dist/{exports_initializers-CYzKLjN7.js → exports_initializers-DIOZQt_L.js}

@@ -1,4 +1,4 @@
-import { R as o } from "./random_width-DN5ZtQkM.js";
+import { R as o } from "./random_width-B_fVXhGx.js";
 function m(r) {
   return new o(r);
 }

package/dist/{floor-BQtb-Azg.js → floor-CymuCmTO.js}

@@ -1,4 +1,4 @@
-import { o as n, q as s, E as t, F as f } from "./index-CUXkjxiT.js";
+import { o as n, n as s, E as t, F as f } from "./index-DSGwv2Yx.js";
 function a(o) {
   const r = { x: s(o, "x", "floor", "float32") };
   return t.runKernel(f, r);

package/dist/{gather-qIqEqaGn.js → gather-DEyjXNb1.js}

@@ -1,4 +1,4 @@
-import { o as g, q as t, E as h, G as p } from "./index-CUXkjxiT.js";
+import { o as g, n as t, E as h, G as p } from "./index-DSGwv2Yx.js";
 function u(n, s, r = 0, e = 0) {
   const o = t(n, "x", "gather"), a = t(s, "indices", "gather", "int32"), c = { x: o, indices: a }, i = { axis: r, batchDims: e };
   return h.runKernel(p, c, i);

package/dist/{gelu-B220X1Go.js → gelu-DpTCC3eB.js}

@@ -1,4 +1,4 @@
-import { e as t, h as n } from "./index-CUXkjxiT.js";
+import { c as t, e as n } from "./index-DSGwv2Yx.js";
 import "./ops/cpu/gelu.js";
 import "./ops/webgl/gelu.js";
 const a = {

package/dist/{gpgpu_math-BwvV12df.js → gpgpu_math-3bCb5ooU.js}

@@ -1,5 +1,5 @@
-import { ab as $, U as E, aR as X, ac as T, aS as ae, K as N, aT as G, x as ie, a9 as k, ad as ce } from "./index-CUXkjxiT.js";
-import { b as se } from "./backend_util-GaFarB78.js";
+import { ab as $, N as E, aR as X, ac as T, aS as ae, I as N, aT as G, v as ie, a9 as k, ad as ce } from "./index-DSGwv2Yx.js";
+import { b as se } from "./backend_util-qwSFfxYx.js";
 const w = {}, M = {
   alpha: !1,
   antialias: !1,
@@ -215,7 +215,7 @@ function Ot(e, t, n) {
 function It(e, t) {
   v(e, () => e.bindFramebuffer(e.FRAMEBUFFER, t)), v(e, () => e.framebufferTexture2D(e.FRAMEBUFFER, e.COLOR_ATTACHMENT0, e.TEXTURE_2D, null, 0));
 }
-function Ut(e) {
+function Lt(e) {
   const t = e.checkFramebufferStatus(e.FRAMEBUFFER);
   if (t !== e.FRAMEBUFFER_COMPLETE)
     throw new Error("Error binding framebuffer: " + Se(e, t));
@@ -258,7 +258,7 @@ function J(e) {
     e[e.length - 1]
   ];
 }
-function Lt(e) {
+function Ut(e) {
   let t = [1, 1, 1];
   return e.length === 0 || e.length === 1 && e[0] === 1 || (t = [Z(e), ...J(e)]), t;
 }
@@ -378,7 +378,7 @@ function Xt(e, t) {
     n != null && ie(n.dtype !== "complex64", () => `${t} does not support complex64 tensors in the WebGL backend.`);
   });
 }
-function U() {
+function L() {
   let e, t, n, r, o, a, i, c, s, l;
   return $().getNumber("WEBGL_VERSION") === 2 ? (e = "#version 300 es", t = "in", n = "out", r = "in", o = "texture", a = "outputColor", i = "out vec4 outputColor;", c = $().getBool("WEBGL2_ISNAN_CUSTOM") ? `
       bool isnan_custom(float val) {
@@ -567,8 +567,8 @@ function Fe(e, t, n) {
   });
   const o = r.join(`
 `), a = e.map((x) => _e(x, t, n.packedInputs, n.enableShapeUniforms)).join(`
-`), i = t.texShape, c = U(), s = Ne(c);
-  let l, u, f = Ue(c);
+`), i = t.texShape, c = L(), s = Ne(c);
+  let l, u, f = Le(c);
   return t.isPacked ? (l = Ae(t.logicalShape, i, n.enableShapeUniforms), u = Ie(c)) : (l = we(t.logicalShape, i, n.enableShapeUniforms), u = Oe(c)), n.packedInputs && (f += be), [
     f,
     s,
@@ -580,7 +580,7 @@ function Fe(e, t, n) {
   ].join(`
 `);
 }
-function L(e, t = !1) {
+function U(e, t = !1) {
   const n = e.shapeInfo.logicalShape;
   switch (n.length) {
     case 0:
@@ -617,7 +617,7 @@ function te(e, t) {
 }
 function _e(e, t, n = !1, r) {
   let o = "";
-  n ? o += te(e, r) : o += L(e, r);
+  n ? o += te(e, r) : o += U(e, r);
   const a = e.shapeInfo.logicalShape, i = t.logicalShape;
   return a.length <= i.length && (n ? o += ot(e, t) : o += at(e, t)), o;
 }
@@ -676,7 +676,7 @@ function Ie(e) {
     }
   `;
 }
-function Ue(e) {
+function Le(e) {
   return `${e.version}
     precision highp float;
     precision highp int;
@@ -732,12 +732,12 @@ function Ue(e) {
       return fract((p3.x + p3.y) * p3.z);
     }
 
-    ${Le}
+    ${Ue}
     ${De}
     ${ge}
   `;
 }
-const Le = `
+const Ue = `
 vec2 uvFromFlat(int texNumR, int texNumC, int index) {
   int texR = index / texNumC;
   int texC = index - texR * texNumC;
@@ -1112,7 +1112,7 @@ function O(e) {
   return `offset${e}`;
 }
 function He(e) {
-  const t = e.name, n = "get" + t.charAt(0).toUpperCase() + t.slice(1), r = U();
+  const t = e.name, n = "get" + t.charAt(0).toUpperCase() + t.slice(1), r = L();
   return `
     vec4 ${n}() {
       return ${r.texture2D}(${t}, halfCR);
@@ -1147,7 +1147,7 @@ function qe(e, t) {
   `;
 }
 function je(e, t) {
-  const n = e.name, r = "get" + n.charAt(0).toUpperCase() + n.slice(1), o = e.shapeInfo.texShape, a = U();
+  const n = e.name, r = "get" + n.charAt(0).toUpperCase() + n.slice(1), o = e.shapeInfo.texShape, a = L();
   if (t)
     return `
     vec4 ${r}(int index) {
@@ -1215,7 +1215,7 @@ function Ye(e, t) {
   `;
 }
 function Ke(e, t) {
-  const n = e.shapeInfo.logicalShape, r = e.name, o = "get" + r.charAt(0).toUpperCase() + r.slice(1), a = e.shapeInfo.texShape, i = a[0], c = a[1], s = U();
+  const n = e.shapeInfo.logicalShape, r = e.name, o = "get" + r.charAt(0).toUpperCase() + r.slice(1), a = e.shapeInfo.texShape, i = a[0], c = a[1], s = L();
   if (a != null && T(n, a))
     return t ? `
       vec4 ${o}(int row, int col) {
@@ -1269,7 +1269,7 @@ function Ze(e, t) {
   if (s.length < n.length) {
     const d = g(e, s), x = ["row", "col"];
     return `
-      ${L(d, t)}
+      ${U(d, t)}
       float ${o}(int row, int col) {
         return ${o}(${b(x, c)});
       }
@@ -1334,7 +1334,7 @@ function Je(e, t) {
         }
       `;
   }
-  const c = U();
+  const c = L();
   if (t)
     return `
     vec4 ${o}(int b, int row, int col) {
@@ -1360,7 +1360,7 @@ function Qe(e, t) {
   if (l.length < n.length) {
     const p = g(e, l), m = ["row", "col", "depth"];
     return `
-        ${L(p, t)}
+        ${U(p, t)}
         float ${o}(int row, int col, int depth) {
           return ${o}(${b(m, s)});
         }
@@ -1430,7 +1430,7 @@ function Qe(e, t) {
   `;
 }
 function et(e, t) {
-  const n = e.name, r = "get" + n.charAt(0).toUpperCase() + n.slice(1), o = U();
+  const n = e.name, r = "get" + n.charAt(0).toUpperCase() + n.slice(1), o = L();
   if (t)
     return `
     vec4 ${r}(int b2, int b, int row, int col) {
@@ -1464,7 +1464,7 @@ function tt(e, t) {
   if (s.length < n.length) {
     const R = g(e, s), F = ["row", "col", "depth", "depth2"];
     return `
-      ${L(R, t)}
+      ${U(R, t)}
       float ${o}(int row, int col, int depth, int depth2) {
         return ${o}(${b(F, l)});
       }
@@ -1550,7 +1550,7 @@ function nt(e) {
   if (s.length < t.length) {
     const p = g(e, s), m = ["row", "col", "depth", "depth2", "depth3"];
     return `
-      ${L(p)}
+      ${U(p)}
       float ${r}(int row, int col, int depth, int depth2, int depth3) {
         return ${r}(${b(m, l)});
       }
@@ -1607,7 +1607,7 @@ function rt(e) {
   if (o.length < t.length) {
     const m = g(e, o), S = ["row", "col", "depth", "depth2", "depth3", "depth4"];
     return `
-      ${L(m)}
+      ${U(m)}
       float ${r}(int row, int col, int depth,
                     int depth2, int depth3, int depth4) {
         return ${r}(${b(S, a)});
@@ -1979,12 +1979,12 @@ export {
   wt as O,
   H as P,
   Nt as Q,
-  Ut as R,
+  Lt as R,
   Ot as S,
   q as T,
   ut as U,
   j as V,
-  Lt as W,
+  Ut as W,
   xt as X,
   ht as Y,
   Yt as Z,
@@ -2008,7 +2008,7 @@ export {
   yt as m,
   Pt as n,
   Vt as o,
-  U as p,
+  L as p,
   Q as q,
   St as r,
   Rt as s,

package/dist/{index-CjOWnMXP.js → index-BQvB7LCC.js}

@@ -1,20 +1,20 @@
-import { W as Mt } from "./backend_webgpu-BqASlsbV.js";
-import { ae as Et, ab as X, x as L, dc as Ut, dd as Ht, bX as Gt, U as D, _ as j, bb as Xt, ai as Ye, aU as Kt, a7 as qt, am as fe, bP as Yt, c7 as jt, c8 as Qt, bV as Zt, cO as Jt, ar as es, ac as De, ah as te, aW as ts, bm as ss, bn as os, bo as is, c9 as as, ca as rs, cb as ns, cc as us, cd as ds, ce as ls, aM as cs, aP as hs, bp as ps, cy as fs, cP as ms, cQ as gs, D as xs, S as Cs, br as ws, bd as ys, de as Ss, aQ as bs, aq as vs, bS as ks, bT as Is, af as Rs, bY as Ps, C as $s, cS as Ds, ao as Ns, z as zs, bt as As, cD as Fs, bu as Ws, cz as Ls, cT as Vs, cA as Bs, bv as Ts, bw as _s, bf as Os, bx as Ms, by as Es, cB as Us, cf as Hs, bz as Gs, cF as Xs, cG as Ks, df as qs, cg as Ys, cU as js, cV as Qs, dg as Zs, c0 as Js, N as eo, be as to, aG as so, cW as oo, bA as io, bB as ao, aN as ro, A as no, bZ as uo, cp as lo, bg as co, F as ho, b_ as po, dh as fo, a9 as at, bs as mo, cE as go, di as xo, al as Co, G as wo, a5 as ke, a$ as yo, b0 as So, cq as bo, ch as vo, ci as ko, cj as Io, aH as Ro, b1 as Po, b2 as $o, dj as Do, aO as No, b3 as zo, b4 as Ao, bD as Fo, cl as Wo, ck as Lo, cY as Vo, b$ as Bo, bE as To, cC as _o, cZ as Oo, c_ as Mo, dk as Eo, aZ as Uo, b5 as Ho, cm as Go, M as Xo, I as Ko, dl as qo, ap as Yo, bi as jo, bj as Qo, bF as Zo, d3 as Jo, bG as ei, P as ti, a6 as si, bH as oi, c$ as ii, aI as ai, c1 as ri, Z as ni, aX as ui, cn as di, H as li, aJ as ci, b9 as hi, d0 as pi, ba as fi, d1 as mi, bJ as gi, bh as xi, b6 as Ci, bK as wi, ak as yi, dm as Si, aL as bi, bL as vi, aF as ki, co as Ii, bM as Ri, bN as Pi, bC as $i, bI as Di, dn as Ni, dp as zi, T as Ai, av as rt, dq as Fi, Q as Wi, J as Li, c3 as Vi, d2 as Bi, b7 as Ti, aK as _i, cr as Oi, dr as Mi, c5 as Ei, cs as Ui, bq as Hi, ds as Gi, ct as Xi, bl as Ki, b8 as qi, bO as Yi, c as ji } from "./index-CUXkjxiT.js";
-import { i as Qi, a as Zi, c as b, f as v, M as Y, b as nt, d as ut, e as dt } from "./webgpu_util-DYlGSwOJ.js";
+import { W as Mt } from "./backend_webgpu-DI2wXEC2.js";
+import { ae as Et, ab as X, v as L, dc as Ut, dd as Ht, bX as Gt, N as D, _ as j, bb as Xt, ai as Ye, aU as Kt, a7 as qt, am as fe, bP as Yt, c7 as jt, c8 as Qt, bV as Zt, cO as Jt, ar as es, ac as De, ah as te, aW as ts, bm as ss, bn as os, bo as is, c9 as as, ca as rs, cb as ns, cc as us, cd as ds, ce as ls, aM as cs, aP as hs, bp as ps, cy as fs, cP as ms, cQ as gs, B as xs, S as Cs, br as ws, bd as ys, de as Ss, aQ as bs, aq as vs, bS as ks, bT as Is, af as Rs, bY as Ps, C as $s, cS as Ds, ao as Ns, x as zs, bt as As, cD as Fs, bu as Ws, cz as Ls, cT as Vs, cA as Bs, bv as Ts, bw as _s, bf as Os, bx as Ms, by as Es, cB as Us, cf as Hs, bz as Gs, cF as Xs, cG as Ks, df as qs, cg as Ys, cU as js, cV as Qs, dg as Zs, c0 as Js, K as eo, be as to, aG as so, cW as oo, bA as io, bB as ao, aN as ro, y as no, bZ as uo, cp as lo, bg as co, F as ho, b_ as po, dh as fo, a9 as at, bs as mo, cE as go, di as xo, al as Co, G as wo, a5 as ke, a$ as yo, b0 as So, cq as bo, ch as vo, ci as ko, cj as Io, aH as Ro, b1 as Po, b2 as $o, dj as Do, aO as No, b3 as zo, b4 as Ao, bD as Fo, cl as Wo, ck as Lo, cY as Vo, b$ as Bo, bE as To, cC as _o, cZ as Oo, c_ as Mo, dk as Eo, aZ as Uo, b5 as Ho, cm as Go, M as Xo, D as Ko, dl as qo, ap as Yo, bi as jo, bj as Qo, bF as Zo, d3 as Jo, bG as ei, P as ti, a6 as si, bH as oi, c$ as ii, aI as ai, c1 as ri, X as ni, aX as ui, cn as di, A as li, aJ as ci, b9 as hi, d0 as pi, ba as fi, d1 as mi, bJ as gi, bh as xi, b6 as Ci, bK as wi, ak as yi, dm as Si, aL as bi, bL as vi, aF as ki, co as Ii, bM as Ri, bN as Pi, bC as $i, bI as Di, dn as Ni, dp as zi, T as Ai, av as rt, dq as Fi, L as Wi, H as Li, c3 as Vi, d2 as Bi, b7 as Ti, aK as _i, cr as Oi, dr as Mi, c5 as Ei, cs as Ui, bq as Hi, ds as Gi, ct as Xi, bl as Ki, b8 as qi, bO as Yi, a as ji } from "./index-DSGwv2Yx.js";
+import { i as Qi, a as Zi, c as b, f as v, M as Y, b as nt, d as ut, e as dt } from "./webgpu_util-DuofJBMo.js";
 import { g as _e, B as F } from "./binary_op_util-pKXltfxI.js";
-import { S as Ji, a as ea } from "./selu_util-BLhIqRkw.js";
-import { E as ta, t as sa, u as oa, w as ia, x as aa, y as ra, f as je, z as lt, A as ct, B as ht, C as na, D as ua, F as da, G as la, H as ca, I as ha, J as pa, K as fa, L as ma, M as ga } from "./backend_util-GaFarB78.js";
-import { t as W, e as S, h as Q, b as G, c as Ie, P as pt, d as xa, a as Ca } from "./webgpu_program-B4HmApL1.js";
-import { r as R, a as wa } from "./Reshape-CPBkTIH2.js";
-import { s as ya } from "./shared-CagdqkLh.js";
-import { c as Oe, a as Ce, b as we, d as Me, e as Sa, g as ft } from "./axis_util-GTVlo58H.js";
-import { h as ye, i as Ne, j as Se, b as Z, d as Ee, g as Ue, k as mt } from "./step-BTxPtq1r.js";
-import { p as ba, a as va, s as ka, b as Ia } from "./slice_util-CC35pLmT.js";
-import { z as Ra } from "./zeros-DvZpK8s6.js";
-import { c as me, a as Pa } from "./concat_util-D18dJ4fD.js";
-import { c as $a, a as Da } from "./segment_util-Dasb2Zaf.js";
+import { S as Ji, a as ea } from "./selu_util-BXdhy_W6.js";
+import { E as ta, t as sa, u as oa, w as ia, x as aa, y as ra, f as je, z as lt, A as ct, B as ht, C as na, D as ua, F as da, G as la, H as ca, I as ha, J as pa, K as fa, L as ma, M as ga } from "./backend_util-qwSFfxYx.js";
+import { t as W, e as S, h as Q, b as G, c as Ie, P as pt, d as xa, a as Ca } from "./webgpu_program-CbjdYLYk.js";
+import { r as R, a as wa } from "./Reshape-pwprEaej.js";
+import { s as ya } from "./shared-DRWDyk9w.js";
+import { c as Oe, a as Ce, b as we, d as Me, e as Sa, g as ft } from "./axis_util-QWWgLjut.js";
+import { h as ye, i as Ne, j as Se, b as Z, d as Ee, g as Ue, k as mt } from "./step-DQY6_ABw.js";
+import { p as ba, a as va, s as ka, b as Ia } from "./slice_util-DPY56GzQ.js";
+import { z as Ra } from "./zeros-Bw0puq_w.js";
+import { c as me, a as Pa } from "./concat_util-C1Mxe27t.js";
+import { c as $a, a as Da } from "./segment_util-4zuHV5IG.js";
 import { n as Na, a as za } from "./non_max_suppression_impl-B2W7YjZB.js";
-import { c as He } from "./scatter_nd_util-ByNJaL6I.js";
+import { c as He } from "./scatter_nd_util-C-x73Cj6.js";
 Qi() && Et(
   "webgpu",
   async () => {

package/dist/{index-CUXkjxiT.js → index-DSGwv2Yx.js}

@@ -3226,34 +3226,34 @@ function vs() {
 vs();
 export {
   di as $,
-  Kr as A,
-  K as B,
+  Ja as A,
+  ur as B,
   yr as C,
-  ur as D,
+  So as D,
   f as E,
   Yr as F,
   Qr as G,
-  Ja as H,
-  So as I,
-  wo as J,
-  Gs as K,
-  Rt as L,
+  wo as H,
+  Gs as I,
+  Rt as J,
+  yo as K,
+  se as L,
   Na as M,
-  yo as N,
+  z as N,
   H as O,
   Oa as P,
-  se as Q,
+  Ue as Q,
   Xa as R,
   uo as S,
   Co as T,
-  z as U,
-  Ue as V,
-  Cs as W,
-  Tn as X,
-  vn as Y,
-  qa as Z,
+  Cs as U,
+  Tn as V,
+  vn as W,
+  qa as X,
+  ni as Y,
+  hs as Z,
   os as _,
-  q as a,
+  Ho as a,
   ea as a$,
   gi as a0,
   tt as a1,
@@ -3318,7 +3318,7 @@ export {
   $s as ax,
   xt as ay,
   he as az,
-  ni as b,
+  q as b,
   Qe as b$,
   na as b0,
   la as b1,
@@ -3383,7 +3383,7 @@ export {
   Fr as bx,
   Mr as by,
   xr as bz,
-  Ho as c,
+  Jo as c,
   en as c$,
   tn as c0,
   za as c1,
@@ -3448,7 +3448,7 @@ export {
   gs as cx,
   cr as cy,
   Tr as cz,
-  M as d,
+  $ as d,
   to as d0,
   Za as d1,
   Bo as d2,
@@ -3491,26 +3491,26 @@ export {
   ja as dx,
   ko as dy,
   Io as dz,
-  Jo as e,
+  ei as e,
   as as f,
   ai as g,
-  ei as h,
-  $ as i,
-  S as j,
+  S as h,
+  nt as i,
+  V as j,
   Fn as k,
-  nt as l,
+  M as l,
   b as m,
-  V as n,
+  T as n,
   N as o,
   Qt as p,
-  T as q,
+  Ct as q,
   ri as r,
   si as s,
   A as t,
-  Ct as u,
-  hs as v,
-  Jn as w,
-  p as x,
-  An as y,
-  wr as z
+  Jn as u,
+  p as v,
+  An as w,
+  wr as x,
+  Kr as y,
+  K as z
 };

package/dist/inference/types.d.ts

@@ -0,0 +1,16 @@
+import { Conversation } from '../tokeniser/type';
+export interface GeneratorConversation extends Conversation {
+    _completed?: boolean;
+    _timestamp?: number;
+}
+export interface GenerateOptions {
+    temperature?: number;
+    topK?: number;
+    topP?: number;
+    usePadding?: boolean;
+    attentionScores?: boolean;
+    includeProbabilities?: boolean;
+    embeddings?: 'embedding' | 'logits' | 'softmax' | 'all';
+    targets?: number[];
+    loraName?: string;
+}

package/dist/inference/types.js

@@ -0,0 +1 @@
+

package/dist/{kernel_funcs_utils-pq0CK9co.js → kernel_funcs_utils-DGqzNlHT.js}

@@ -1,6 +1,6 @@
-import { _ as B, U as G, aU as K, a7 as W, aH as z, aV as V, ab as N, aI as F, am as S } from "./index-CUXkjxiT.js";
-import { u as O, f as H } from "./gpgpu_math-BwvV12df.js";
-import { f as v } from "./backend_util-GaFarB78.js";
+import { _ as B, N as G, aU as K, a7 as W, aH as z, aV as V, ab as N, aI as F, am as S } from "./index-DSGwv2Yx.js";
+import { u as O, f as H } from "./gpgpu_math-3bCb5ooU.js";
+import { f as v } from "./backend_util-qwSFfxYx.js";
 function Y(t, e) {
   return ["x", "y", "z", "w", "u", "v"].slice(0, e).map((s) => `${t}.${s}`);
 }
@@ -200,12 +200,12 @@ const be = {
   backendName: "webgl",
   kernelFunc: oe
 };
-const U = "return (a < 0.) ? b * a : a;", k = `
+const k = "return (a < 0.) ? b * a : a;", U = `
   vec4 aLessThanZero = vec4(lessThan(a, vec4(0.)));
   return (aLessThanZero * (b * a)) + ((vec4(1.0) - aLessThanZero) * a);
 `;
 function ue(t) {
-  const { inputs: e, backend: s } = t, { x: r, alpha: u } = e, n = N().getBool("WEBGL_PACK_BINARY_OPERATIONS") ? new E(k, r.shape, u.shape) : new b(U, r.shape, u.shape);
+  const { inputs: e, backend: s } = t, { x: r, alpha: u } = e, n = N().getBool("WEBGL_PACK_BINARY_OPERATIONS") ? new E(U, r.shape, u.shape) : new b(k, r.shape, u.shape);
   return s.runWebGLProgram(n, [r, u], "float32");
 }
 const Ne = {
@@ -273,7 +273,7 @@ function Ae(t, e = !1) {
   if (t === "relu6")
     return e ? ae : Q;
   if (t === "prelu")
-    return e ? k : U;
+    return e ? U : k;
   if (t === "leakyrelu")
     return e ? R : w;
   if (t === "sigmoid")

package/dist/layers/BaseLayer.js

@@ -1,4 +1,4 @@
-import { a2 as p, h as s, a4 as g } from "../index-CUXkjxiT.js";
+import { a2 as p, e as s, a4 as g } from "../index-DSGwv2Yx.js";
 import b from "./WeightStore.js";
 class T {
   parent;
@@ -59,11 +59,11 @@ class T {
   checkpointingFn(t, ...e) {
     const r = this.trainableVariables;
     return p((...i) => {
-      const o = i[i.length - 1], a = i.slice(0, e.length), h = this.forward(t, ...a);
-      return o(a), { value: h, gradFunc: (n, l) => {
+      const o = i[i.length - 1], a = i.slice(0, e.length), n = this.forward(t, ...a);
+      return o(a), { value: n, gradFunc: (h, l) => {
         const c = s().state.activeTape;
         s().state.activeTape = [];
-        const d = g((...u) => this.forward(t, ...u.slice(0, a.length)))([...l, ...r], n);
+        const d = g((...u) => this.forward(t, ...u.slice(0, a.length)))([...l, ...r], h);
         return s().state.activeTape = c, d;
       } };
     })(...e, ...r);

package/dist/layers/CausalSelfAttention.js

@@ -1,11 +1,11 @@
 import { attentionMask as R } from "../ops/attentionMask.js";
 import J from "./BaseLayer.js";
-import { r as v } from "../rope-0j_f1TPm.js";
+import { r as v } from "../rope-CC5RjmKU.js";
 import { appendCache as A } from "../ops/appendCache.js";
-import { k as c, t as L } from "../index-CUXkjxiT.js";
+import { k as c, t as L } from "../index-DSGwv2Yx.js";
 import { softmax16 as y } from "../ops/softmax16.js";
-import { b as M } from "../matMul16-BcVC_E62.js";
-import { p as K } from "../pack16-bqltoUlR.js";
+import { b as M } from "../matMul16-BIT70Vya.js";
+import { p as K } from "../pack16-WlOSOuZA.js";
 import { transpose16 as j } from "../ops/transpose16.js";
 import { dot16 as E } from "../ops/dot16.js";
 import { reshape16 as _ } from "../ops/reshape16.js";
@@ -13,8 +13,8 @@ import { isPackedTensor as l } from "../utilities/packed.js";
 import { qkv as q } from "../ops/qkv.js";
 import { normRMS as O } from "../ops/normRMS.js";
 import { dropout16 as x } from "../ops/dropout16.js";
-import { v as P } from "../variable-Ck482e3n.js";
-import { r as S } from "../random_normal-IBRrha8a.js";
+import { v as P } from "../variable-SSATClyt.js";
+import { r as S } from "../random_normal-CIm8lk2-.js";
 class it extends J {
   constructor(t, o, s, i) {
     super(o, i), this.attentionConfig = s, this.index = t, this.units = o.nEmbed * 3, this.projUnits = o.nEmbed, this.ATTN = `block_${this.index}_cAttn`, this.PROJ = `block_${this.index}_cProj`, this.addVariable(this.ATTN), this.addVariable(this.PROJ), this.divisor = 1 / Math.sqrt(o.nEmbed / o.nHead);

package/dist/layers/LoRA.js

@@ -1,8 +1,8 @@
-import { a as m, t as n } from "../index-CUXkjxiT.js";
+import { b as m, t as n } from "../index-DSGwv2Yx.js";
 import { p } from "../index-DmeWGGmS.js";
-import { v as g } from "../variable-Ck482e3n.js";
-import { r as S } from "../random_normal-IBRrha8a.js";
-import { z as _ } from "../zeros-DvZpK8s6.js";
+import { v as g } from "../variable-SSATClyt.js";
+import { r as S } from "../random_normal-CIm8lk2-.js";
+import { z as _ } from "../zeros-Bw0puq_w.js";
 class B {
   weightStore;
   alpha;

package/dist/layers/MLP.js

@@ -1,10 +1,10 @@
-import { t as M } from "../index-CUXkjxiT.js";
+import { t as M } from "../index-DSGwv2Yx.js";
 import f from "./BaseLayer.js";
-import { b as h } from "../matMul16-BcVC_E62.js";
+import { b as h } from "../matMul16-BIT70Vya.js";
 import { reshape16 as d } from "../ops/reshape16.js";
 import { dropout16 as L } from "../ops/dropout16.js";
-import { v as n } from "../variable-Ck482e3n.js";
-import { r as m } from "../random_normal-IBRrha8a.js";
+import { v as n } from "../variable-SSATClyt.js";
+import { r as m } from "../random_normal-CIm8lk2-.js";
 class N extends f {
   index;
   hiddenUnits;

package/dist/layers/PositionEmbedding.js

@@ -1,9 +1,9 @@
-import { t as c, an as p, a as s } from "../index-CUXkjxiT.js";
+import { t as c, an as p, b as s } from "../index-DSGwv2Yx.js";
 import f from "./BaseLayer.js";
-import { E as u } from "../random_width-DN5ZtQkM.js";
-import { r as b } from "../exports_initializers-CYzKLjN7.js";
-import { m as g } from "../mod-CSdCpRjf.js";
-import { r as l } from "../range-C-CjF-LI.js";
+import { E as u } from "../random_width-B_fVXhGx.js";
+import { r as b } from "../exports_initializers-DIOZQt_L.js";
+import { m as g } from "../mod-BXjLYwvM.js";
+import { r as l } from "../range-BDxO73mk.js";
 function h(e) {
   return new u(e);
 }

package/dist/layers/RMSNorm.js

@@ -1,8 +1,8 @@
-import { t as i } from "../index-CUXkjxiT.js";
+import { t as i } from "../index-DSGwv2Yx.js";
 import e from "./BaseLayer.js";
 import { normRMS as m } from "../ops/normRMS.js";
-import { v as a } from "../variable-Ck482e3n.js";
-import { o as M } from "../ones-BytntneX.js";
+import { v as a } from "../variable-SSATClyt.js";
+import { o as M } from "../ones-DbVB5N58.js";
 class l extends e {
   GAMMA;
   rmsConfig;

package/dist/layers/RoPECache.js

@@ -1,7 +1,7 @@
-import { i as a, a as n, p as c, t as p, k as r } from "../index-CUXkjxiT.js";
-import { r as l } from "../zeros_like-CWjDdwr-.js";
-import { c as f, s as m } from "../unsorted_segment_sum-DJvk5xnh.js";
-import { r as h } from "../range-C-CjF-LI.js";
+import { d as a, b as n, p as c, t as p, k as r } from "../index-DSGwv2Yx.js";
+import { r as l } from "../zeros_like-rOHr54NY.js";
+import { c as f, s as m } from "../unsorted_segment_sum-BfFVV9Zm.js";
+import { r as h } from "../range-BDxO73mk.js";
 class x {
   constructor(s) {
     this.config = s;