@genai-fi/nanogpt 0.10.3 → 0.12.0

package/dist/{dataset-BcwmTGYc.js → dataset-CVIJu7Xa.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, a3 as h, aa as N, d as v, aD as o, aE as p, aF as g, x as k, t as y } from "./index-Duu1Lvvv.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-CDWShFHF.js";
+import { t as B } from "./tensor-CEt9Nm2s.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;
   }

package/dist/{dropout-BcvN9JYi.js → dropout-DLhSMNTZ.js}

@@ -1,7 +1,7 @@
-import { A as l, C as h, at as d, n as f, B as m, l as u, a8 as p, w as g, x as V, m as v } from "./index-DOvlwCh-.js";
-import { f as w } from "./floor-D5QdR_le.js";
+import { o as l, u as h, at as d, ab as f, q as m, x as u, a3 as p, j as g, l as V, m as v } from "./index-Duu1Lvvv.js";
+import { f as w } from "./floor-BRMPgeIs.js";
 import { s as c } from "./index-CieiGp4Y.js";
-class N {
+class b {
   constructor(t, e, s, n, o) {
     this.mean = t, this.stdDev = e, this.dtype = s, this.nextVal = NaN, this.truncated = n, this.truncated && (this.upper = this.mean + this.stdDev * 2, this.lower = this.mean - this.stdDev * 2);
     const a = o || Math.random();
@@ -33,7 +33,7 @@ class N {
     return t <= this.upper && t >= this.lower;
   }
 }
-class b {
+class N {
   constructor(t = 0, e = 1, s, n) {
     if (this.canReturnFloat = () => this.dtype == null || this.dtype === "float32", this.min = t, this.range = e - t, this.dtype = s, n == null && (n = Math.random()), typeof n == "number" && (n = n.toString()), !this.canReturnFloat() && this.range <= 1)
       throw new Error(`The difference between ${t} - ${e} <= 1 and dtype is not float`);
@@ -49,7 +49,7 @@ class b {
 function T(r, t = 0, e = 1, s, n) {
   if (h(r), s != null && s === "bool")
     throw new Error(`Unsupported data type ${s}`);
-  const o = new N(t, e, s, !1, n), a = d(r, s);
+  const o = new b(t, e, s, !1, n), a = d(r, s);
   for (let i = 0; i < a.values.length; i++)
     a.values[i] = o.nextValue();
   return a.toTensor();
@@ -57,7 +57,7 @@ function T(r, t = 0, e = 1, s, n) {
 const U = /* @__PURE__ */ l({ randomNormal_: T });
 function x(r, t = 0, e = 1, s = "float32", n) {
   h(r);
-  const o = d(r, s), a = new b(t, e, null, n);
+  const o = d(r, s), a = new N(t, e, null, n);
   for (let i = 0; i < o.values.length; i++)
     o.values[i] = a.nextValue();
   return o.toTensor();
@@ -83,10 +83,10 @@ function D(r, t, e, s) {
   const o = M(n, e), a = 1 - t, i = g(w(V(y(o, 0, 1, "float32", s), a)), a);
   return v(n, i);
 }
-const F = /* @__PURE__ */ l({ dropout_: D });
+const q = /* @__PURE__ */ l({ dropout_: D });
 export {
-  N as M,
+  b as M,
   y as a,
-  F as d,
+  q as d,
   U as r
 };

package/dist/expand_dims-ChkuOp6I.js

@@ -0,0 +1,11 @@
+import { o as e, q as a, x as i, E as m, A as p } from "./index-Duu1Lvvv.js";
+function x(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 });
+export {
+  u as e
+};

package/dist/{exports_initializers-Hta_rEnm.js → exports_initializers-1KWPiStI.js}

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

package/dist/{floor-D5QdR_le.js → floor-BRMPgeIs.js}

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

package/dist/{gather-D3JcZUaI.js → gather-BSULDalH.js}

@@ -1,4 +1,4 @@
-import { A as g, B as t, E as h, K as p } from "./index-DOvlwCh-.js";
+import { o as g, q as t, E as h, G as p } from "./index-Duu1Lvvv.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-CjNPL4OH.js → gelu-BK1k-n1i.js}

@@ -1,4 +1,4 @@
-import { u as t, e as n } from "./index-DOvlwCh-.js";
+import { i as t, e as n } from "./index-Duu1Lvvv.js";
 import "./ops/cpu/gelu.js";
 import "./ops/webgl/gelu.js";
 const a = {

package/dist/{gpgpu_math-DAOmgtXR.js → gpgpu_math-BJSTk_mW.js}

@@ -1,5 +1,5 @@
-import { j as $, h as E, aN as X, n as T, aO as ae, V as N, aP as G, l as ie, ad as k, o as ce } from "./index-DOvlwCh-.js";
-import { b as se } from "./backend_util-RCe-rHaj.js";
+import { aa as $, U as E, aN as X, ab as T, aO as ae, K as N, aP as G, x as ie, a8 as k, ac as ce } from "./index-Duu1Lvvv.js";
+import { b as se } from "./backend_util-DC3rBo_H.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 Lt(e) {
+function Ut(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 Ut(e) {
+function Lt(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 L() {
+function U() {
   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 = L(), s = Ne(c);
-  let l, u, f = Le(c);
+`), i = t.texShape, c = U(), s = Ne(c);
+  let l, u, f = Ue(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 U(e, t = !1) {
+function L(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 += U(e, r);
+  n ? o += te(e, r) : o += L(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 Le(e) {
+function Ue(e) {
   return `${e.version}
     precision highp float;
     precision highp int;
@@ -732,12 +732,12 @@ function Le(e) {
       return fract((p3.x + p3.y) * p3.z);
     }
 
-    ${Ue}
+    ${Le}
     ${De}
     ${ge}
   `;
 }
-const Ue = `
+const Le = `
 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 = L();
+  const t = e.name, n = "get" + t.charAt(0).toUpperCase() + t.slice(1), r = U();
   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 = L();
+  const n = e.name, r = "get" + n.charAt(0).toUpperCase() + n.slice(1), o = e.shapeInfo.texShape, a = U();
   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 = L();
+  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();
   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 `
-      ${U(d, t)}
+      ${L(d, t)}
       float ${o}(int row, int col) {
         return ${o}(${b(x, c)});
       }
@@ -1334,7 +1334,7 @@ function Je(e, t) {
         }
       `;
   }
-  const c = L();
+  const c = U();
   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 `
-        ${U(p, t)}
+        ${L(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 = L();
+  const n = e.name, r = "get" + n.charAt(0).toUpperCase() + n.slice(1), o = U();
   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 `
-      ${U(R, t)}
+      ${L(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 `
-      ${U(p)}
+      ${L(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 `
-      ${U(m)}
+      ${L(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,
-  Lt as R,
+  Ut as R,
   Ot as S,
   q as T,
   ut as U,
   j as V,
-  Ut as W,
+  Lt 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,
-  L as p,
+  U as p,
   Q as q,
   St as r,
   Rt as s,