@genai-fi/nanogpt 0.5.1 → 0.5.2

package/dist/{mat_mul-D0SifYfJ.js → mat_mul-DeGU1U_C.js}

@@ -1,4 +1,4 @@
-import { o as m, i as s, q as c, E as M, B as p } from "./index-iNhkcAEQ.js";
+import { o as m, j as s, u as c, E as M, B as p } from "./index-CnHyhpKc.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -21,7 +21,7 @@ function f(e, o, n = !1, l = !1) {
   const r = { a, b: t }, u = { transposeA: n, transposeB: l };
   return M.runKernel(p, r, u);
 }
-const b = /* @__PURE__ */ m({ matMul_: f });
+const i = /* @__PURE__ */ m({ matMul_: f });
 export {
-  b as m
+  i as m
 };

package/dist/{max-CYaAjEEp.js → max-CcnEArWK.js}

@@ -1,4 +1,4 @@
-import { o as r, i as e, E as x, M as c } from "./index-iNhkcAEQ.js";
+import { o as r, j as e, E as x, M as c } from "./index-CnHyhpKc.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -15,11 +15,11 @@ import { o as r, i as e, E as x, M as c } from "./index-iNhkcAEQ.js";
  * limitations under the License.
  * =============================================================================
  */
-function i(n, o = null, s = !1) {
+function m(n, o = null, s = !1) {
   const t = { x: e(n, "x", "max") }, a = { reductionIndices: o, keepDims: s };
   return x.runKernel(c, t, a);
 }
-const l = /* @__PURE__ */ r({ max_: i });
+const l = /* @__PURE__ */ r({ max_: m });
 export {
   l as m
 };

package/dist/{moments-B06NlR_V.js → moments-DLTE6-1p.js}

@@ -1,6 +1,6 @@
-import { o as m, i as c, E as i, a0 as f, a1 as l, a2 as h, s as x, y as d } from "./index-iNhkcAEQ.js";
-import { e as v } from "./axis_util-97KkkyRQ.js";
-import { r as E } from "./reshape-DxTPgnwL.js";
+import { o as m, j as c, E as f, a0 as i, a1 as l, a2 as h, s as x, D as d } from "./index-CnHyhpKc.js";
+import { e as v } from "./axis_util-CVbf1vmL.js";
+import { r as E } from "./reshape-CTIbqjwm.js";
 /**
  * @license
  * Copyright 2020 Google Inc. All Rights Reserved.
@@ -19,7 +19,7 @@ import { r as E } from "./reshape-DxTPgnwL.js";
  */
 function S(a, t = null, e = !1) {
   const s = { x: c(a, "x", "mean") }, o = { axis: t, keepDims: e };
-  return i.runKernel(f, s, o);
+  return f.runKernel(i, s, o);
 }
 const r = /* @__PURE__ */ m({ mean_: S });
 /**

package/dist/{norm-D3676xIo.js → norm-BpWsOapl.js}

@@ -1,8 +1,8 @@
-import { o as l, i as c, E as y, a3 as E, a1 as w, a4 as o, a5 as u, U as v, f as I, a2 as A } from "./index-iNhkcAEQ.js";
-import { e as $ } from "./axis_util-97KkkyRQ.js";
-import { m as f } from "./max-CYaAjEEp.js";
-import { r as h } from "./reshape-DxTPgnwL.js";
-import { s as t } from "./sum-B_92TaHD.js";
+import { o as l, j as c, E as y, a3 as E, a1 as w, a4 as o, a5 as u, W as v, f as I, a2 as A } from "./index-CnHyhpKc.js";
+import { e as $ } from "./axis_util-CVbf1vmL.js";
+import { m as f } from "./max-CcnEArWK.js";
+import { r as h } from "./reshape-CTIbqjwm.js";
+import { s as t } from "./sum-UdfvaNhB.js";
 /**
  * @license
  * Copyright 2020 Google Inc. All Rights Reserved.

package/dist/{ones-BIeFnPHR.js → ones-CDWGzVnm.js}

@@ -1,5 +1,5 @@
-import { l as n, n as t, p as m, E as i } from "./index-iNhkcAEQ.js";
-import { z as l, c } from "./zeros-NMYTayy7.js";
+import { n, p as t, q as m, E as i } from "./index-CnHyhpKc.js";
+import { z as c, c as f } from "./zeros-CYMicyqz.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -16,14 +16,14 @@ import { z as l, c } from "./zeros-NMYTayy7.js";
  * limitations under the License.
  * =============================================================================
  */
-function f(o, r = "float32") {
+function l(o, r = "float32") {
   if (n(o), r === "complex64") {
-    const s = f(o, "float32"), a = l(o, "float32");
-    return c(s, a);
+    const s = l(o, "float32"), a = c(o, "float32");
+    return f(s, a);
   }
   const e = t(m(o), r);
   return i.makeTensor(e, o, r);
 }
 export {
-  f as o
+  l as o
 };

package/dist/ops/appendCache.js

@@ -1,8 +1,8 @@
-import { e as a } from "../index-iNhkcAEQ.js";
+import { e as a } from "../index-CnHyhpKc.js";
 import "./cpu/appendCache.js";
 import "./webgl/appendCache.js";
-import { c as s } from "../concat-Cxbo2sOz.js";
-import { z as c } from "../zeros-NMYTayy7.js";
+import { c as s } from "../concat-BRRtq4S2.js";
+import { z as c } from "../zeros-CYMicyqz.js";
 function i(r, p, n, o) {
   if (!o) {
     const e = r.shape[2];

package/dist/ops/attentionMask.js

@@ -1,4 +1,4 @@
-import { e as o } from "../index-iNhkcAEQ.js";
+import { e as o } from "../index-CnHyhpKc.js";
 import "./cpu/attentionMask.js";
 import "./webgl/attentionMask.js";
 import "./grads/attentionMask.js";

package/dist/ops/cpu/appendCache.js

@@ -1,5 +1,5 @@
-import { r as d } from "../../index-iNhkcAEQ.js";
-import { c as h } from "../../concat-Cxbo2sOz.js";
+import { r as d } from "../../index-CnHyhpKc.js";
+import { c as h } from "../../concat-BRRtq4S2.js";
 function u(p) {
   const { cache: n, item: s } = p.inputs, { maxSize: r, pastLen: c } = p.attrs, t = n.shape[0], o = n.shape[1], a = n.shape[3], e = s.shape[2];
   if (c + e <= r) {

package/dist/ops/cpu/attentionMask.js

@@ -1,8 +1,8 @@
-import { r as a, g as p, f as u } from "../../index-iNhkcAEQ.js";
-import { l as N, w as b } from "../../ops-ObfXLHYQ.js";
-import { o as g } from "../../ones-BIeFnPHR.js";
-import { z as A } from "../../zeros-NMYTayy7.js";
-import { m as I } from "../../mat_mul-D0SifYfJ.js";
+import { r as a, g as p, f as u } from "../../index-CnHyhpKc.js";
+import { l as N, w as b } from "../../ops-DzQTmLIl.js";
+import { o as g } from "../../ones-CDWGzVnm.js";
+import { z as A } from "../../zeros-CYMicyqz.js";
+import { m as I } from "../../mat_mul-DeGU1U_C.js";
 function o(n) {
   const { q: s, k: e } = n.inputs, { divisor: r } = n.attrs, c = s.shape[2], t = e.shape[2], m = N.bandPart(g([t, t]), -1, 0).cast("bool"), l = A([t, t]), i = p([t, t], Number.NEGATIVE_INFINITY), f = b(m, l, i), k = I(s, e, !1, !0).mul(u(r)), d = f.slice([0, 0], [c, t]).expandDims(0).expandDims(0);
   return k.add(d);

package/dist/ops/cpu/fusedSoftmax.js

@@ -1,5 +1,5 @@
-import { r as n } from "../../index-iNhkcAEQ.js";
-import { s as f } from "../../softmax-BjsptB07.js";
+import { r as n } from "../../index-CnHyhpKc.js";
+import { s as f } from "../../softmax-DX6qXAbm.js";
 function r(t) {
   const { inputs: s, attrs: i } = t, { logits: o } = s, { dim: a, dropoutRate: e } = i;
   if (!o)

package/dist/ops/cpu/gatherSub.js

@@ -1,6 +1,6 @@
-import { o as u, i as c, E as g, L as h, r as m, s as p } from "../../index-iNhkcAEQ.js";
-import { r as l } from "../../range-BsFU-SNG.js";
-import { s as N } from "../../stack--cqr9Dgc.js";
+import { o as u, j as c, E as g, O as h, r as m, s as p } from "../../index-CnHyhpKc.js";
+import { r as l } from "../../range-CkOJ7090.js";
+import { s as N } from "../../stack-S2-D2JAQ.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -23,8 +23,8 @@ function f(e, s) {
 }
 const b = /* @__PURE__ */ u({ gatherND_: f });
 function d(e) {
-  const { values: s, labels: n, logits: r } = e.inputs, t = n.shape[0], i = l(0, t, 1, "int32"), o = N([i, n], 1), a = b(r, o);
-  return p(s, a);
+  const { values: s, labels: n, logits: r } = e.inputs, t = n.shape[0], o = l(0, t, 1, "int32"), a = N([o, n], 1), i = b(r, a);
+  return p(s, i);
 }
 const k = {
   kernelName: "EfficientGatherSub",

package/dist/ops/cpu/gelu.js

@@ -1,4 +1,4 @@
-import { r as t, t as d } from "../../index-iNhkcAEQ.js";
+import { r as t, t as d } from "../../index-CnHyhpKc.js";
 const o = 0.7978845608028654, c = 0.044715;
 function m(u) {
   const { inputs: l } = u, { x: e } = l, n = e;

package/dist/ops/cpu/matMulGelu.js

@@ -1,4 +1,4 @@
-import { r as a, t as i } from "../../index-iNhkcAEQ.js";
+import { r as a, t as i } from "../../index-CnHyhpKc.js";
 const c = 0.7978845608028654, m = 0.044715;
 function M(o) {
   const { inputs: s } = o, { x: t, kernel: l } = s, e = t, u = l;

package/dist/ops/cpu/matMulMul.js

@@ -1,4 +1,4 @@
-import { r as n, t as M } from "../../index-iNhkcAEQ.js";
+import { r as n, t as M } from "../../index-CnHyhpKc.js";
 function e(t) {
   const { inputs: r, attrs: o } = t, { transposeA: s, transposeB: l } = o, { x: c, kernel: u, y: a } = r, m = c, i = u, k = a;
   return M(() => m.matMul(i, s, l).mul(k));

package/dist/ops/cpu/mulDropout.js

@@ -1,4 +1,4 @@
-import { r as e, b as u } from "../../index-iNhkcAEQ.js";
+import { r as e, b as u } from "../../index-CnHyhpKc.js";
 function n(o) {
   const { inputs: r } = o, { a: l, b: t } = r;
   return console.warn("Using fallback mulDrop implementation without dropout."), u(l, t);

package/dist/ops/cpu/normRMS.js

@@ -1,4 +1,4 @@
-import { r as o, t as d } from "../../index-iNhkcAEQ.js";
+import { r as o, t as d } from "../../index-CnHyhpKc.js";
 function i(t) {
   const { inputs: e } = t, { x: n, gamma: s } = e, r = n, a = s;
   return d(() => {

package/dist/ops/cpu/qkv.js

@@ -1,6 +1,6 @@
-import { r as q } from "../../index-iNhkcAEQ.js";
-import { r as o } from "../../reshape-DxTPgnwL.js";
-import { s as x } from "../../split-BCbrzthj.js";
+import { r as q } from "../../index-CnHyhpKc.js";
+import { r as o } from "../../reshape-CTIbqjwm.js";
+import { s as x } from "../../split-CVwhL8Oe.js";
 function v(p) {
   const { x: c, kernel: K } = p.inputs, { heads: n } = p.attrs, [s, e, t] = c.shape, a = o(c, [s * e, t]), i = a.dot(K);
   a.dispose();

package/dist/ops/cpu/rope.js

@@ -1,8 +1,8 @@
-import { r as S } from "../../index-iNhkcAEQ.js";
-import { r as F } from "../../range-BsFU-SNG.js";
-import { g as I } from "../../gather-Bxe1Qip8.js";
-import { s as E } from "../../stack--cqr9Dgc.js";
-import { c as T } from "../../concat-Cxbo2sOz.js";
+import { r as S } from "../../index-CnHyhpKc.js";
+import { r as F } from "../../range-CkOJ7090.js";
+import { g as I } from "../../gather-BWyutxwi.js";
+import { s as E } from "../../stack-S2-D2JAQ.js";
+import { c as T } from "../../concat-BRRtq4S2.js";
 function U(t, c, p, o, r) {
   const n = o.shape[3], s = p;
   if (s > n) return o;

package/dist/ops/cpu/scatterSub.js

@@ -1,27 +1,27 @@
-import { o as l, l as g, i, E as k, a6 as w, r as $, s as d, b as m } from "../../index-iNhkcAEQ.js";
-import { r as b } from "../../range-BsFU-SNG.js";
-import { s as E } from "../../stack--cqr9Dgc.js";
-import { o as D } from "../../ones-BIeFnPHR.js";
-function N(n, r, t) {
-  const s = r.rank > 1 ? r.shape[r.rank - 1] : 1, e = r.rank > 1 ? r.rank - 1 : 1, o = `Must have updates.shape = indices.shape[:batchDim] + shape[sliceDim:], got updates.shape: ${t.shape}, indices.shape: ${r.shape}, shape: ${n}, sliceDim: ${s}, and batchDim: ${e}.`;
+import { o as l, n as g, j as h, E as k, a6 as w, r as $, s as d, b as m } from "../../index-CnHyhpKc.js";
+import { r as b } from "../../range-CkOJ7090.js";
+import { s as E } from "../../stack-S2-D2JAQ.js";
+import { o as D } from "../../ones-CDWGzVnm.js";
+function N(a, r, t) {
+  const s = r.rank > 1 ? r.shape[r.rank - 1] : 1, e = r.rank > 1 ? r.rank - 1 : 1, o = `Must have updates.shape = indices.shape[:batchDim] + shape[sliceDim:], got updates.shape: ${t.shape}, indices.shape: ${r.shape}, shape: ${a}, sliceDim: ${s}, and batchDim: ${e}.`;
   if (t.rank < e)
     throw new Error(o + ` update.rank < ${e}. `);
-  if (n.length < s + (t.rank - e))
+  if (a.length < s + (t.rank - e))
     throw new Error(o + ` Output shape length < ${s + (t.rank - e)}`);
-  if (t.rank !== e + n.length - s)
-    throw new Error(o + ` update.rank != ${e + n.length - s}`);
-  for (let a = 0; a < e; ++a)
-    if (t.shape[a] !== r.shape[a])
-      throw new Error(o + ` updates.shape[${a}] (${t.shape[a]}) != indices.shape[${a}] (${r.shape[a]}).`);
-  for (let a = 0; a < t.rank - e; ++a)
-    if (t.shape[a + e] !== n[a + s])
-      throw new Error(o + ` updates.shape[${a + e}] (${t.shape[a + e]}) != shape[${a + e}] (${n[a + e]})`);
+  if (t.rank !== e + a.length - s)
+    throw new Error(o + ` update.rank != ${e + a.length - s}`);
+  for (let n = 0; n < e; ++n)
+    if (t.shape[n] !== r.shape[n])
+      throw new Error(o + ` updates.shape[${n}] (${t.shape[n]}) != indices.shape[${n}] (${r.shape[n]}).`);
+  for (let n = 0; n < t.rank - e; ++n)
+    if (t.shape[n + e] !== a[n + s])
+      throw new Error(o + ` updates.shape[${n + e}] (${t.shape[n + e]}) != shape[${n + e}] (${a[n + e]})`);
 }
-function S(n, r, t) {
+function S(a, r, t) {
   if (r.rank < 1)
     throw new Error(`tf.scatterND() expects the indices to be rank 1 or higher, but the rank was ${r.rank}.`);
-  if (n.rank < 1)
-    throw new Error(`tf.scatterND() expects the updates to be rank 1 or higher, but the rank was ${n.rank}.`);
+  if (a.rank < 1)
+    throw new Error(`tf.scatterND() expects the updates to be rank 1 or higher, but the rank was ${a.rank}.`);
   if (r.dtype !== "int32")
     throw new Error(`The dtype of 'indices' should be int32, but got dtype: ${r.dtype}`);
   if (t.length < 1)
@@ -29,10 +29,10 @@ function S(n, r, t) {
   if (t.length === 0) {
     if (r.size === 0)
       throw new Error(`Indices specified for empty output. indices shape: ${r.shape}`);
-    if (n.size === 0)
-      throw new Error(`Updates specified for empty output. updates shape: ${n.shape}`);
+    if (a.size === 0)
+      throw new Error(`Updates specified for empty output. updates shape: ${a.shape}`);
   }
-  N(t, r, n);
+  N(t, r, a);
 }
 /**
  * @license
@@ -50,16 +50,16 @@ function S(n, r, t) {
  * limitations under the License.
  * =============================================================================
  */
-function y(n, r, t) {
+function y(a, r, t) {
   g(t);
-  const s = i(n, "indices", "scatterND", "int32"), e = i(r, "updates", "scatterND");
+  const s = h(a, "indices", "scatterND", "int32"), e = h(r, "updates", "scatterND");
   S(e, s, t);
-  const o = { indices: s, updates: e }, a = { shape: t };
-  return k.runKernel(w, o, a);
+  const o = { indices: s, updates: e }, n = { shape: t };
+  return k.runKernel(w, o, n);
 }
 const v = /* @__PURE__ */ l({ scatterND_: y });
-function I(n) {
-  const { logits: r, labels: t, dy: s } = n.inputs, e = t.shape[0], o = r.shape[1], a = b(0, e, 1, "int32"), h = E([a, t], 1), c = D([e]), p = v(h, c, [e, o]), f = d(r, p), u = s.reshape([e, 1]);
+function I(a) {
+  const { logits: r, labels: t, dy: s } = a.inputs, e = t.shape[0], o = r.shape[1], n = b(0, e, 1, "int32"), i = E([n, t], 1), c = D([e]), p = v(i, c, [e, o]), f = d(r, p), u = s.reshape([e, 1]);
   return m(f, u);
 }
 const T = {

package/dist/ops/fusedSoftmax.js

@@ -1,4 +1,4 @@
-import { e as t } from "../index-iNhkcAEQ.js";
+import { e as t } from "../index-CnHyhpKc.js";
 import "./cpu/fusedSoftmax.js";
 import "./webgl/fusedSoftmax.js";
 import "./grads/fusedSoftmax.js";

package/dist/ops/gatherSub.js

@@ -1,4 +1,4 @@
-import { e as n } from "../index-iNhkcAEQ.js";
+import { e as n } from "../index-CnHyhpKc.js";
 import "./cpu/gatherSub.js";
 import "./webgl/gatherSub.js";
 function f(r, e, t) {

package/dist/ops/gelu.js

@@ -1,4 +1,4 @@
-import "../index-iNhkcAEQ.js";
+import "../index-CnHyhpKc.js";
 import "./cpu/gelu.js";
 import "./webgl/gelu.js";
 import { d as e, g as i } from "./grads/gelu.js";

package/dist/ops/grads/attentionMask.js

@@ -1,4 +1,4 @@
-import { h as l, f as a } from "../../index-iNhkcAEQ.js";
+import { h as l, f as a } from "../../index-CnHyhpKc.js";
 import { matMulMul as i } from "../matMulMul.js";
 const m = {
   kernelName: "AttentionMask",

package/dist/ops/grads/fusedSoftmax.js

@@ -1,6 +1,6 @@
-import { h as d, b as u, s as f } from "../../index-iNhkcAEQ.js";
+import { h as d, b as u, s as f } from "../../index-CnHyhpKc.js";
 import { mulDrop as l } from "../mulDrop.js";
-import { s as g } from "../../sum-B_92TaHD.js";
+import { s as g } from "../../sum-UdfvaNhB.js";
 const T = {
   kernelName: "FusedSoftmax",
   outputsToSave: [!0],

package/dist/ops/grads/gelu.js

@@ -1,4 +1,4 @@
-import { h as t, e as n } from "../../index-iNhkcAEQ.js";
+import { h as t, e as n } from "../../index-CnHyhpKc.js";
 import "../cpu/gelu.js";
 import "../webgl/gelu.js";
 const o = {

package/dist/ops/grads/matMulGelu.js

@@ -1,4 +1,4 @@
-import { h as a, e as o } from "../../index-iNhkcAEQ.js";
+import { h as a, e as o } from "../../index-CnHyhpKc.js";
 function s(e, n, r) {
   return o().runKernel("MatMulGeluGrad", { dy: e, x: n, kernel: r });
 }

package/dist/ops/grads/normRMS.js

@@ -1,4 +1,4 @@
-import { h as t, e as g } from "../../index-iNhkcAEQ.js";
+import { h as t, e as g } from "../../index-CnHyhpKc.js";
 function s(r, a, n) {
   return g().runKernel("RMSNormGrad", { dy: r, x: a, gamma: n });
 }

package/dist/ops/grads/qkv.js

@@ -1,4 +1,4 @@
-import { h as Q } from "../../index-iNhkcAEQ.js";
+import { h as Q } from "../../index-CnHyhpKc.js";
 const V = {
   kernelName: "QKV",
   inputsToSave: ["x", "kernel"],

package/dist/ops/grads/rope.js

@@ -1,4 +1,4 @@
-import { h as a, e as i } from "../../index-iNhkcAEQ.js";
+import { h as a, e as i } from "../../index-CnHyhpKc.js";
 function p(n, e, s, o) {
   return i().runKernel("Rope", { x: n, sin: e, cos: s }, { pastLen: o });
 }

package/dist/ops/matMulGelu.js

@@ -1,4 +1,4 @@
-import { e as u } from "../index-iNhkcAEQ.js";
+import { e as u } from "../index-CnHyhpKc.js";
 import "./cpu/matMulGelu.js";
 import "./webgl/matMulGelu.js";
 import "./grads/matMulGelu.js";

package/dist/ops/matMulMul.js

@@ -1,4 +1,4 @@
-import { e as u } from "../index-iNhkcAEQ.js";
+import { e as u } from "../index-CnHyhpKc.js";
 import "./cpu/matMulMul.js";
 import "./webgl/matMulMul.js";
 function m(e, r, t, l = !1, n = !1) {

package/dist/ops/mulDrop.js

@@ -1,4 +1,4 @@
-import { e as t } from "../index-iNhkcAEQ.js";
+import { e as t } from "../index-CnHyhpKc.js";
 import "./cpu/mulDropout.js";
 import "./webgl/mulDropout.js";
 function m(r, o, e, n) {

package/dist/ops/node/sparseCrossEntropy.js

@@ -1,4 +1,4 @@
-import { r as o } from "../../index-iNhkcAEQ.js";
+import { r as o } from "../../index-CnHyhpKc.js";
 function r(e) {
   const { logits: t, labels: n } = e.inputs;
   return e.backend.executeMultipleOutputs("SparseSoftmaxCrossEntropyWithLogits", [], [t, n], 2);

package/dist/ops/normRMS.js

@@ -1,4 +1,4 @@
-import { e as n } from "../index-iNhkcAEQ.js";
+import { e as n } from "../index-CnHyhpKc.js";
 import "./cpu/normRMS.js";
 import "./webgl/normRMS.js";
 import "./grads/normRMS.js";

package/dist/ops/qkv.js

@@ -1,4 +1,4 @@
-import { e as o } from "../index-iNhkcAEQ.js";
+import { e as o } from "../index-CnHyhpKc.js";
 import "./cpu/qkv.js";
 import "./webgl/qkv.js";
 import "./grads/qkv.js";

package/dist/ops/scatterSub.js

@@ -1,4 +1,4 @@
-import { e as i } from "../index-iNhkcAEQ.js";
+import { e as i } from "../index-CnHyhpKc.js";
 import "./cpu/scatterSub.js";
 import "./webgl/scatterSub.js";
 function c(t, r, e) {

package/dist/ops/webgl/appendCache.js

@@ -1,4 +1,4 @@
-import { r as p } from "../../index-iNhkcAEQ.js";
+import { r as p } from "../../index-CnHyhpKc.js";
 class m {
   variableNames = ["cache", "item"];
   outputShape;

package/dist/ops/webgl/attentionMask.js

@@ -1,4 +1,4 @@
-import { r as m } from "../../index-iNhkcAEQ.js";
+import { r as m } from "../../index-CnHyhpKc.js";
 class h {
   variableNames = ["q", "k"];
   outputShape;