@genai-fi/nanogpt 0.10.1 → 0.10.2

package/dist/{gpgpu_math-DDVJCn6-.js → gpgpu_math-DsCcikas.js}

@@ -1,7 +1,7 @@
-import { K as z, L as oe, M as ie, N as ae, P as me, Q as J, h as $e } from "./index-ZyQhjEPo.js";
-import { e as v, a as F, i as B, h as I, b as T, s as E, J as H, K as Te, G as Y } from "./tensor-DdQUJZlz.js";
+import { K as z, L as oe, M as ie, N as ae, P as me, Q as J, h as $e } from "./index-D6Q1lPZO.js";
+import { e as v, a as F, i as B, h as I, b as T, s as E, I as H, J as Te, F as Y } from "./tensor-CzmOBsdf.js";
 import "./index-Cp39cXWe.js";
-import { b as Se } from "./backend_util-D-rUb2ty.js";
+import { b as Se } from "./backend_util-B1XRLuq9.js";
 const Re = "model", Ee = ".json", Ce = ".weights.bin";
 function K(e) {
   return new Promise((t) => setTimeout(t)).then(e);

package/dist/{index-ZyQhjEPo.js → index-D6Q1lPZO.js}

@@ -1,5 +1,5 @@
-import { w as J, o as je, p as ge, K as qe, I as Ee, q as pe, s as ye, t as Te, v as Ve, a3 as He, x as Ae, A as Me, a4 as Ne, a5 as xe, m as G, a6 as Xe, $ as Je, a7 as Ye, a8 as Ze, a9 as Qe, aa as et, ab as tt, ac as nt, ad as st, ae as rt, af as at } from "./tensor_util-DV-FP5Q3.js";
-import { o as it, E as ot, q as ct, y as lt, a as g, r as ut, t as ht, T as R, u as dt, V as re, v as ae, w as Y, x as be, f as ft, m as mt, s as Z, e as B, M as gt, N as ue, D, O as Fe, B as De, P as pt, d as he, Q as yt, R as bt, S as wt, U as St, j as Ce } from "./tensor-DdQUJZlz.js";
+import { w as J, o as je, p as ge, K as qe, I as Ee, q as pe, s as ye, t as Te, v as Ve, a3 as He, x as Ae, A as Me, a4 as Ne, a5 as xe, m as G, a6 as Xe, $ as Je, a7 as Ye, a8 as Ze, a9 as Qe, aa as et, ab as tt, ac as nt, ad as st, ae as rt, af as at } from "./tensor_util-DfwaWayG.js";
+import { L as it, E as ot, w as ct, x as lt, a as g, o as ut, q as ht, T as R, r as dt, V as re, t as ae, u as Y, v as be, f as ft, m as mt, s as Z, e as B, M as gt, N as ue, C, O as Fe, A as Ce, P as pt, d as he, Q as yt, R as bt, S as wt, U as St, j as De } from "./tensor-CzmOBsdf.js";
 import { p as Q } from "./index-xuotMAFm.js";
 import { B as ee } from "./index-Cp39cXWe.js";
 function ne(s) {
@@ -602,7 +602,7 @@ E.registerFlag("CANVAS2D_WILL_READ_FREQUENTLY_FOR_GPU", () => !1);
 E.registerFlag("USE_SETTIMEOUTCUSTOM", () => !1);
 function Et(s, e) {
   let t = s;
-  if (D(s))
+  if (C(s))
     return e === "string" ? [] : [s.length];
   if (Ne(s)) {
     const r = s.channels || "RGBA";
@@ -612,12 +612,12 @@ function Et(s, e) {
   if (!Array.isArray(s))
     return [];
   const n = [];
-  for (; Array.isArray(t) || D(t) && e !== "string"; )
+  for (; Array.isArray(t) || C(t) && e !== "string"; )
     n.push(t.length), t = t[0];
   return Array.isArray(s) && B().getBool("TENSORLIKE_CHECK_SHAPE_CONSISTENCY") && $e(s, n, []), n;
 }
 function $e(s, e, t) {
-  if (t = t || [], !Array.isArray(s) && !D(s)) {
+  if (t = t || [], !Array.isArray(s) && !C(s)) {
     g(e.length === 0, () => `Element arr[${t.join("][")}] is a primitive, but should be an array/TypedArray of ${e[0]} elements`);
     return;
   }
@@ -638,13 +638,13 @@ function S(s, e, t, n = "numeric") {
   if (s instanceof gt())
     return Se(n, s.dtype, e, t), s;
   let r = ue(s);
-  if (r !== "string" && ["bool", "int32", "float32"].indexOf(n) >= 0 && (r = n), Se(n, r, e, t), s == null || !D(s) && !Array.isArray(s) && typeof s != "number" && typeof s != "boolean" && typeof s != "string") {
+  if (r !== "string" && ["bool", "int32", "float32"].indexOf(n) >= 0 && (r = n), Se(n, r, e, t), s == null || !C(s) && !Array.isArray(s) && typeof s != "number" && typeof s != "boolean" && typeof s != "string") {
     const c = s == null ? "null" : s.constructor.name;
     throw new Error(`Argument '${e}' passed to '${t}' must be a Tensor or TensorLike, but got '${c}'`);
   }
   const a = Et(s, r);
-  !D(s) && !Array.isArray(s) && (s = [s]);
-  const o = r !== "string" ? Fe(s, r) : De(s, [], !0);
+  !C(s) && !Array.isArray(s) && (s = [s]);
+  const o = r !== "string" ? Fe(s, r) : Ce(s, [], !0);
   return u.makeTensor(o, a, r);
 }
 function Tt(s, e, t, n = "numeric") {
@@ -681,7 +681,7 @@ function Mt(s, e, t, n) {
       throw new Error(`Creating tensor from GPU data only supports 'float32'|'int32' dtype, while the dtype is ${n}.`);
     return u.backend.createTensorFromGPUData(s, e || t, n);
   }
-  if (!D(s) && !Array.isArray(s) && typeof s != "number" && typeof s != "boolean" && typeof s != "string")
+  if (!C(s) && !Array.isArray(s) && typeof s != "number" && typeof s != "boolean" && typeof s != "string")
     throw new Error("values passed to tensor(values) must be a number/boolean/string or an array of numbers/booleans/strings, or a TypedArray");
   if (e != null) {
     he(e);
@@ -692,7 +692,7 @@ function Mt(s, e, t, n) {
       g(t[i] === e[i] || !c, () => `Error creating a new Tensor. Inferred shape (${t}) does not match the provided shape (${e}). `);
     }
   }
-  return !D(s) && !Array.isArray(s) && (s = [s]), e = e || t, s = n !== "string" ? Fe(s, n) : De(s, [], !0), u.makeTensor(s, e, n);
+  return !C(s) && !Array.isArray(s) && (s = [s]), e = e || t, s = n !== "string" ? Fe(s, n) : Ce(s, [], !0), u.makeTensor(s, e, n);
 }
 class H {
   /**
@@ -706,7 +706,7 @@ class H {
     return new H(e).slice();
   }
   constructor(e) {
-    if (this.shards = [], this.previousShardIndex = 0, e == null || (e instanceof Array || (e = [e]), e = e.map((n) => D(n) ? n.buffer : n), e.length === 0))
+    if (this.shards = [], this.previousShardIndex = 0, e == null || (e instanceof Array || (e = [e]), e = e.map((n) => C(n) ? n.buffer : n), e.length === 0))
       return;
     this.bufferUniformSize = e[0].byteLength;
     let t = 0;
@@ -765,10 +765,10 @@ function Nt(s, e) {
 function _n() {
   u.disposeVariables();
 }
-function On() {
+function Ln() {
   return u;
 }
-function Ln() {
+function On() {
   return u.memory();
 }
 function k(s, e) {
@@ -840,7 +840,7 @@ const de = typeof ee < "u" && (typeof Blob > "u" || typeof atob > "u" || typeof
 function Ie(s) {
   return de ? ee.byteLength(s, "utf8") : new Blob([s]).size;
 }
-function Dt(s) {
+function Ct(s) {
   if (de)
     return ee.from(s).toString("base64");
   const e = new Uint8Array(s);
@@ -849,7 +849,7 @@ function Dt(s) {
     t += String.fromCharCode(e[n]);
   return btoa(t);
 }
-function Ct(s) {
+function Dt(s) {
   if (de) {
     const n = ee.from(s, "base64");
     return n.buffer.slice(n.byteOffset, n.byteOffset + n.byteLength);
@@ -965,8 +965,8 @@ class v {
   }
 }
 const Xn = (s) => v.getSaveHandlers(s);
-const ie = "tensorflowjs", oe = 1, O = "models_store", $ = "model_info_store";
-function Oe() {
+const ie = "tensorflowjs", oe = 1, L = "models_store", $ = "model_info_store";
+function Le() {
   if (!B().getBool("IS_BROWSER"))
     throw new Error("Failed to obtain IndexedDB factory because the current environmentis not a web browser.");
   const s = typeof window > "u" ? self : window, e = s.indexedDB || s.mozIndexedDB || s.webkitIndexedDB || s.msIndexedDB || s.shimIndexedDB;
@@ -976,11 +976,11 @@ function Oe() {
 }
 function ce(s) {
   const e = s.result;
-  e.createObjectStore(O, { keyPath: "modelPath" }), e.createObjectStore($, { keyPath: "modelPath" });
+  e.createObjectStore(L, { keyPath: "modelPath" }), e.createObjectStore($, { keyPath: "modelPath" });
 }
 class z {
   constructor(e) {
-    if (this.indexedDB = Oe(), e == null || !e)
+    if (this.indexedDB = Le(), e == null || !e)
       throw new Error("For IndexedDB, modelPath must not be null, undefined or empty.");
     this.modelPath = e;
   }
@@ -1012,7 +1012,7 @@ class z {
       a.onupgradeneeded = () => ce(a), a.onsuccess = () => {
         const i = a.result;
         if (t == null) {
-          const o = i.transaction(O, "readonly"), l = o.objectStore(O).get(this.modelPath);
+          const o = i.transaction(L, "readonly"), l = o.objectStore(L).get(this.modelPath);
           l.onsuccess = () => {
             if (l.result == null)
               return i.close(), r(new Error(`Cannot find model with path '${this.modelPath}' in IndexedDB.`));
@@ -1029,8 +1029,8 @@ class z {
           }
           let m;
           h.onsuccess = () => {
-            m = i.transaction(O, "readwrite");
-            const d = m.objectStore(O);
+            m = i.transaction(L, "readwrite");
+            const d = m.objectStore(L);
             let p;
             try {
               p = d.put({
@@ -1055,18 +1055,18 @@ class z {
   }
 }
 z.URL_SCHEME = "indexeddb://";
-const Le = (s) => B().getBool("IS_BROWSER") && !Array.isArray(s) && s.startsWith(z.URL_SCHEME) ? $t(s.slice(z.URL_SCHEME.length)) : null;
-v.registerSaveRouter(Le);
-v.registerLoadRouter(Le);
+const Oe = (s) => B().getBool("IS_BROWSER") && !Array.isArray(s) && s.startsWith(z.URL_SCHEME) ? $t(s.slice(z.URL_SCHEME.length)) : null;
+v.registerSaveRouter(Oe);
+v.registerLoadRouter(Oe);
 function $t(s) {
   return new z(s);
 }
 function _t(s) {
   return s.startsWith(z.URL_SCHEME) ? s.slice(z.URL_SCHEME.length) : s;
 }
-class Ot {
+class Lt {
   constructor() {
-    this.indexedDB = Oe();
+    this.indexedDB = Le();
   }
   async listModels() {
     return new Promise((e, t) => {
@@ -1093,8 +1093,8 @@ class Ot {
             return a.close(), n(new Error(`Cannot find model with path '${e}' in IndexedDB.`));
           {
             const h = o.delete(e), m = () => {
-              l = a.transaction(O, "readwrite");
-              const p = l.objectStore(O).delete(e);
+              l = a.transaction(L, "readwrite");
+              const p = l.objectStore(L).delete(e);
               p.onsuccess = () => t(c.result.modelArtifactsInfo), p.onerror = (b) => n(c.error);
             };
             h.onsuccess = m, h.onerror = (d) => (m(), a.close(), n(c.error));
@@ -1106,14 +1106,14 @@ class Ot {
     });
   }
 }
-const C = "/", K = "tensorflowjs_models", ze = "info", Lt = "model_topology", zt = "weight_specs", Pt = "weight_data", Ut = "model_metadata";
+const D = "/", K = "tensorflowjs_models", ze = "info", Ot = "model_topology", zt = "weight_specs", Pt = "weight_data", Ut = "model_metadata";
 function Pe(s) {
   return {
-    info: [K, s, ze].join(C),
-    topology: [K, s, Lt].join(C),
-    weightSpecs: [K, s, zt].join(C),
-    weightData: [K, s, Pt].join(C),
-    modelMetadata: [K, s, Ut].join(C)
+    info: [K, s, ze].join(D),
+    topology: [K, s, Ot].join(D),
+    weightSpecs: [K, s, zt].join(D),
+    weightData: [K, s, Pt].join(D),
+    modelMetadata: [K, s, Ut].join(D)
   };
 }
 function Ue(s) {
@@ -1121,10 +1121,10 @@ function Ue(s) {
     window.localStorage.removeItem(e);
 }
 function Gt(s) {
-  const e = s.split(C);
+  const e = s.split(D);
   if (e.length < 3)
     throw new Error(`Invalid key format: ${s}`);
-  return e.slice(1, e.length - 1).join(C);
+  return e.slice(1, e.length - 1).join(D);
 }
 function Wt(s) {
   return s.startsWith(P.URL_SCHEME) ? s.slice(P.URL_SCHEME.length) : s;
@@ -1152,7 +1152,7 @@ class P {
     {
       const t = JSON.stringify(e.modelTopology), n = JSON.stringify(e.weightSpecs), r = _e(e), a = H.join(e.weightData);
       try {
-        this.LS.setItem(this.keys.info, JSON.stringify(r)), this.LS.setItem(this.keys.topology, t), this.LS.setItem(this.keys.weightSpecs, n), this.LS.setItem(this.keys.weightData, Dt(a));
+        this.LS.setItem(this.keys.info, JSON.stringify(r)), this.LS.setItem(this.keys.topology, t), this.LS.setItem(this.keys.weightSpecs, n), this.LS.setItem(this.keys.weightData, Ct(a));
         const i = {
           format: e.format,
           generatedBy: e.generatedBy,
@@ -1199,7 +1199,7 @@ class P {
     const i = this.LS.getItem(this.keys.weightData);
     if (i == null)
       throw new Error(`In local storage, the binary weight values of model '${this.modelPath}' are missing.`);
-    return t.weightData = Ct(i), t;
+    return t.weightData = Dt(i), t;
   }
 }
 P.URL_SCHEME = "localstorage://";
@@ -1214,7 +1214,7 @@ class jt {
     g(B().getBool("IS_BROWSER"), () => "Current environment is not a web browser"), g(typeof window > "u" || typeof window.localStorage < "u", () => "Current browser does not appear to support localStorage"), this.LS = window.localStorage;
   }
   async listModels() {
-    const e = {}, t = K + C, n = C + ze;
+    const e = {}, t = K + D, n = D + ze;
     for (let r = 0; r < this.LS.length; ++r) {
       const a = this.LS.key(r);
       if (a.startsWith(t) && a.endsWith(n)) {
@@ -1310,7 +1310,7 @@ if (B().get("IS_BROWSER")) {
   } catch {
   }
   try {
-    N.registerManager(z.URL_SCHEME, new Ot());
+    N.registerManager(z.URL_SCHEME, new Lt());
   } catch {
   }
 }
@@ -1464,9 +1464,9 @@ function dn(s, e) {
 }
 const Be = /* @__PURE__ */ M({ pow_: dn });
 function U(s, e) {
-  if ((D(s) && e !== "string" || Array.isArray(s)) && e !== "complex64")
+  if ((C(s) && e !== "string" || Array.isArray(s)) && e !== "complex64")
     throw new Error("Error creating a new Scalar: value must be a primitive (number|boolean|string)");
-  if (e === "string" && D(s) && !(s instanceof Uint8Array))
+  if (e === "string" && C(s) && !(s instanceof Uint8Array))
     throw new Error("When making a scalar from encoded string, the value must be `Uint8Array`.");
   return Mt(s, [], [], e);
 }
@@ -1479,14 +1479,14 @@ function mn(s) {
   const e = S(s, "x", "square"), t = {};
   return u.runKernel("Square", { x: e }, t);
 }
-const L = /* @__PURE__ */ M({ square_: mn });
+const O = /* @__PURE__ */ M({ square_: mn });
 function Zn(s) {
   return g(Y(s), () => "The f passed in grads(f) must be a function"), (e, t) => {
     g(Array.isArray(e), () => "The args passed in grads(f)(args) must be an array of `Tensor`s or `TensorLike`s");
     const n = Tt(e, "args", "tf.grads", "string_or_numeric"), r = t != null ? S(t, "dy", "tf.grads") : null;
     return u.tidy(() => {
       const { value: a, grads: i } = u.gradients(() => s(...n), n, r);
-      return r != null && Ce(a.shape, r.shape, "The shape of dy passed in grads(f)([x1,...], dy) must match the shape returned by f([x1,...])"), We(i), i;
+      return r != null && De(a.shape, r.shape, "The shape of dy passed in grads(f)([x1,...], dy) must match the shape returned by f([x1,...])"), We(i), i;
     });
   };
 }
@@ -1494,7 +1494,7 @@ function Qn(s) {
   return g(Y(s), () => "The f passed in valueAndGrads(f) must be a function"), (e, t) => {
     g(Array.isArray(e) && e.every((r) => r instanceof R), () => "The args passed in valueAndGrads(f)(args) must be array of tensors"), g(t == null || t instanceof R, () => "The dy passed in valueAndGrads(f)(args, dy) must be a tensor");
     const n = u.gradients(() => s(...e), e, t);
-    return t != null && Ce(n.value.shape, t.shape, "The shape of dy passed in valueAndGrads(f)([x1,...], dy) must match the shape returned by f([x1,...])"), We(n.grads), n;
+    return t != null && De(n.value.shape, t.shape, "The shape of dy passed in valueAndGrads(f)([x1,...], dy) must match the shape returned by f([x1,...])"), We(n.grads), n;
   };
 }
 function gn(s, e) {
@@ -1692,7 +1692,7 @@ class vn extends W {
         return;
       const c = this.accumulatedGrads[r].variable, l = this.accumulatedUpdates[r].variable;
       k(() => {
-        const h = y(f(c, this.rho), f(L(o), 1 - this.rho)), m = f(x(V(y(l, this.epsilon)), V(y(c, this.epsilon))), o), d = y(f(l, this.rho), f(L(m), 1 - this.rho));
+        const h = y(f(c, this.rho), f(O(o), 1 - this.rho)), m = f(x(V(y(l, this.epsilon)), V(y(c, this.epsilon))), o), d = y(f(l, this.rho), f(O(m), 1 - this.rho));
         c.assign(h), l.assign(d);
         const p = y(f(m, -this.learningRate), a);
         a.assign(p);
@@ -1749,7 +1749,7 @@ class Bn extends W {
         return;
       const o = this.accumulatedGrads[r].variable;
       k(() => {
-        const c = y(o, L(i));
+        const c = y(o, O(i));
         o.assign(c);
         const l = y(f(x(i, V(y(c, u.backend.epsilon()))), -this.learningRate), a);
         a.assign(l);
@@ -1804,7 +1804,7 @@ class En extends W {
         const l = Array.isArray(e) ? e[i].tensor : e[a];
         if (l == null)
           return;
-        const h = this.accumulatedFirstMoment[i].variable, m = this.accumulatedSecondMoment[i].variable, d = y(f(h, this.beta1), f(l, 1 - this.beta1)), p = y(f(m, this.beta2), f(L(l), 1 - this.beta2)), b = x(d, n), w = x(p, r);
+        const h = this.accumulatedFirstMoment[i].variable, m = this.accumulatedSecondMoment[i].variable, d = y(f(h, this.beta1), f(l, 1 - this.beta1)), p = y(f(m, this.beta2), f(O(l), 1 - this.beta2)), b = x(d, n), w = x(p, r);
         h.assign(d), m.assign(p);
         const I = y(f(x(b, y(V(w), this.epsilon)), -this.learningRate), o);
         o.assign(I);
@@ -2026,14 +2026,14 @@ class Mn extends W {
         return;
       const c = this.accumulatedMeanSquares[r].variable, l = this.accumulatedMoments[r].variable;
       k(() => {
-        const h = y(f(c, this.decay), f(L(o), 1 - this.decay));
+        const h = y(f(c, this.decay), f(O(o), 1 - this.decay));
         if (this.centered) {
-          const m = this.accumulatedMeanGrads[r].variable, d = y(f(m, this.decay), f(o, 1 - this.decay)), p = x(f(o, this.learningRate), V(j(h, y(L(d), this.epsilon)))), b = y(f(l, this.momentum), p);
+          const m = this.accumulatedMeanGrads[r].variable, d = y(f(m, this.decay), f(o, 1 - this.decay)), p = x(f(o, this.learningRate), V(j(h, y(O(d), this.epsilon)))), b = y(f(l, this.momentum), p);
           c.assign(h), m.assign(d), l.assign(b);
           const w = j(a, b);
           a.assign(w);
         } else {
-          const m = y(f(c, this.decay), f(L(o), 1 - this.decay)), d = y(f(l, this.momentum), x(f(o, this.learningRate), V(y(m, this.epsilon))));
+          const m = y(f(c, this.decay), f(O(o), 1 - this.decay)), d = y(f(l, this.momentum), x(f(o, this.learningRate), V(y(m, this.epsilon))));
           c.assign(m), l.assign(d);
           const p = j(a, d);
           a.assign(p);
@@ -2111,7 +2111,7 @@ export {
   Hn as Q,
   Wn as R,
   cn as S,
-  L as T,
+  O as T,
   Ke as U,
   An as V,
   Mn as W,
@@ -2119,7 +2119,7 @@ export {
   vn as Y,
   Tn as Z,
   Bn as _,
-  Ln as a,
+  On as a,
   In as a0,
   kn as a1,
   _ as a2,
@@ -2132,7 +2132,7 @@ export {
   U as b,
   j as c,
   A as d,
-  On as e,
+  Ln as e,
   Gn as f,
   Un as g,
   Jn as h,

package/dist/{index-CjOj7j-u.js → index-DRyE072i.js}

@@ -1,20 +1,20 @@
-import { W as Mt } from "./backend_webgpu-B0u2ndUn.js";
-import { f as Et, j, J as ke } from "./index-ZyQhjEPo.js";
-import { i as Ut, a as Ht, c as b, f as v, M as Y, b as at, d as rt, e as nt } from "./webgpu_util-BBCnKm2X.js";
-import { e as X, a as L, N as Gt, Z as Xt, s as D, l as Ye, b as De, p as te, Y as Kt, g as qt, i as ut, j as Yt, A as dt, f as jt } from "./tensor-DdQUJZlz.js";
+import { W as Mt } from "./backend_webgpu-CqpfEImu.js";
+import { f as Et, j, J as ke } from "./index-D6Q1lPZO.js";
+import { i as Ut, a as Ht, c as b, f as v, M as Y, b as at, d as rt, e as nt } from "./webgpu_util-0_ubCEHJ.js";
+import { e as X, a as L, N as Gt, Z as Xt, s as D, l as Ye, b as De, p as te, Y as Kt, g as qt, i as ut, j as Yt, z as dt, f as jt } from "./tensor-CzmOBsdf.js";
 import { g as _e, B as F } from "./binary_op_util-pKXltfxI.js";
-import { S as Qt, a as Zt, h as Ce, i as Ne, j as we, d as Q, e as Oe, g as Me, k as lt } from "./selu_util-D1w6yyTO.js";
-import { E as Jt, t as es, u as ts, w as ss, x as os, y as is, f as je, z as ct, A as ht, B as pt, C as as, D as rs, F as ns, G as us, H as ds, I as ls, J as cs, K as hs, L as ps, M as fs, N as ms, O as gs } from "./backend_util-D-rUb2ty.js";
-import { t as W, e as S, h as Z, b as G, c as Ie, P as ft, d as xs, a as Cs } from "./webgpu_program-Cigz-7RF.js";
-import { aa as ws, a2 as ys, I as Ss, h as bs, u as fe, a9 as vs, bi as ks, bj as Is, A as Rs, bk as Ps, H as $s, _ as Ds, aM as Ns, aN as zs, aO as As, bl as Fs, bm as Ws, bn as Ls, bp as Vs, bo as Bs, bq as Ts, y as _s, aq as Os, aP as Ms, aQ as Es, br as Us, bs as Hs, B as Gs, f as Xs, aS as Ks, ag as qs, bN as Ys, as as js, F as Qs, x as Zs, aG as Js, a1 as eo, a8 as to, D as so, C as oo, aU as io, be as ao, aV as ro, aW as no, bu as uo, aX as lo, l as co, aY as ho, aw as po, aZ as fo, a_ as mo, a$ as go, bO as xo, b0 as Co, bg as wo, bh as yo, bP as So, bv as bo, bw as vo, bx as ko, bQ as Io, a7 as Ro, g as Po, ai as $o, V as Do, by as No, aH as zo, b1 as Ao, z as Fo, E as Wo, aI as Lo, bR as Vo, ax as Bo, Q as To, a6 as _o, bS as Oo, aT as Mo, bf as Eo, bT as Uo, k as Ho, G as Go, ak as Xo, al as Ko, bU as qo, bz as Yo, bA as jo, bB as Qo, W as Zo, am as Jo, an as ei, bV as ti, L as si, ao as oi, ap as ii, b3 as ai, bW as ri, bE as ni, bD as ui, af as di, b4 as li, b5 as ci, bF as hi, bG as pi, bX as fi, aj as mi, ar as gi, bH as xi, M as Ci, S as wi, i as yi, N as Si, az as bi, aA as vi, b6 as ki, ab as Ii, b7 as Ri, P as Pi, b8 as $i, ac as Di, X as Ni, aJ as zi, R as Ai, $ as Fi, d as Wi, e as Li, Y as Vi, aC as Bi, bI as Ti, aD as _i, bJ as Oi, ba as Mi, ay as Ei, at as Ui, aK as Hi, j as Gi, bY as Xi, ah as Ki, bb as qi, U as Yi, bK as ji, n as Qi, bc as Zi, b2 as Ji, b9 as ea, bZ as ta, b_ as sa, T as oa, b$ as ia, c as aa, ad as ra, bL as na, au as ua, Z as da, c0 as la, c1 as ca, ae as ha, bM as pa, aR as fa, c2 as ma, c3 as ga, aE as xa, av as Ca, bd as wa, r as ya } from "./tensor_util-DV-FP5Q3.js";
-import { r as R, a as Sa } from "./Reshape-CF6odzV4.js";
-import { s as ba } from "./shared-BuAXb4CI.js";
-import { c as Ee, a as ye, b as Se, d as Ue, e as va, g as mt } from "./axis_util-BvHEw88j.js";
-import { p as ka, a as Ia, b as Ra, d as Pa } from "./slice_util-DtEldBfK.js";
-import { z as $a } from "./zeros-2gldETuK.js";
-import { c as me, a as Da } from "./concat_util-DcJk7YHS.js";
+import { S as Qt, a as Zt, h as Ce, i as Ne, j as we, d as Q, e as Oe, g as Me, k as lt } from "./selu_util-BGPXmd4B.js";
+import { E as Jt, t as es, u as ts, w as ss, x as os, y as is, f as je, z as ct, A as ht, B as pt, C as as, D as rs, F as ns, G as us, H as ds, I as ls, J as cs, K as hs, L as ps, M as fs, N as ms, O as gs } from "./backend_util-B1XRLuq9.js";
+import { t as W, e as S, h as Z, b as G, c as Ie, P as ft, d as xs, a as Cs } from "./webgpu_program-DzaQiqel.js";
+import { aa as ws, a2 as ys, I as Ss, h as bs, u as fe, a9 as vs, bi as ks, bj as Is, A as Rs, bk as Ps, H as $s, _ as Ds, aM as Ns, aN as zs, aO as As, bl as Fs, bm as Ws, bn as Ls, bp as Vs, bo as Bs, bq as Ts, y as _s, aq as Os, aP as Ms, aQ as Es, br as Us, bs as Hs, B as Gs, f as Xs, aS as Ks, ag as qs, bN as Ys, as as js, F as Qs, x as Zs, aG as Js, a1 as eo, a8 as to, D as so, C as oo, aU as io, be as ao, aV as ro, aW as no, bu as uo, aX as lo, l as co, aY as ho, aw as po, aZ as fo, a_ as mo, a$ as go, bO as xo, b0 as Co, bg as wo, bh as yo, bP as So, bv as bo, bw as vo, bx as ko, bQ as Io, a7 as Ro, g as Po, ai as $o, V as Do, by as No, aH as zo, b1 as Ao, z as Fo, E as Wo, aI as Lo, bR as Vo, ax as Bo, Q as To, a6 as _o, bS as Oo, aT as Mo, bf as Eo, bT as Uo, k as Ho, G as Go, ak as Xo, al as Ko, bU as qo, bz as Yo, bA as jo, bB as Qo, W as Zo, am as Jo, an as ei, bV as ti, L as si, ao as oi, ap as ii, b3 as ai, bW as ri, bE as ni, bD as ui, af as di, b4 as li, b5 as ci, bF as hi, bG as pi, bX as fi, aj as mi, ar as gi, bH as xi, M as Ci, S as wi, i as yi, N as Si, az as bi, aA as vi, b6 as ki, ab as Ii, b7 as Ri, P as Pi, b8 as $i, ac as Di, X as Ni, aJ as zi, R as Ai, $ as Fi, d as Wi, e as Li, Y as Vi, aC as Bi, bI as Ti, aD as _i, bJ as Oi, ba as Mi, ay as Ei, at as Ui, aK as Hi, j as Gi, bY as Xi, ah as Ki, bb as qi, U as Yi, bK as ji, n as Qi, bc as Zi, b2 as Ji, b9 as ea, bZ as ta, b_ as sa, T as oa, b$ as ia, c as aa, ad as ra, bL as na, au as ua, Z as da, c0 as la, c1 as ca, ae as ha, bM as pa, aR as fa, c2 as ma, c3 as ga, aE as xa, av as Ca, bd as wa, r as ya } from "./tensor_util-DfwaWayG.js";
+import { r as R, a as Sa } from "./Reshape-CDVLyVfz.js";
+import { s as ba } from "./shared-D2NP_CpY.js";
+import { c as Ee, a as ye, b as Se, d as Ue, e as va, g as mt } from "./axis_util-BsIr9ZNu.js";
+import { p as ka, a as Ia, b as Ra, d as Pa } from "./slice_util-DyjSAD0u.js";
+import { z as $a } from "./zeros-DBFVbpv5.js";
+import { c as me, a as Da } from "./concat_util-iBYIyuQe.js";
 import { n as Na, a as za } from "./non_max_suppression_impl-B2W7YjZB.js";
-import { c as He } from "./scatter_nd_util-5EL-8VAQ.js";
+import { c as He } from "./scatter_nd_util-C7zXRT_h.js";
 Ut() && Et(
   "webgpu",
   async () => {

package/dist/{kernel_funcs_utils-Dg_-E44D.js → kernel_funcs_utils-CWfOAPGO.js}

@@ -1,8 +1,8 @@
-import { j as B } from "./index-ZyQhjEPo.js";
-import { u as N, f as G } from "./gpgpu_math-DDVJCn6-.js";
-import { s as K, I as W, e as O } from "./tensor-DdQUJZlz.js";
-import { I as z, h as V, W as F, X as Y, u as S } from "./tensor_util-DV-FP5Q3.js";
-import { f as v } from "./backend_util-D-rUb2ty.js";
+import { j as B } from "./index-D6Q1lPZO.js";
+import { u as N, f as G } from "./gpgpu_math-DsCcikas.js";
+import { s as K, H as W, e as O } from "./tensor-CzmOBsdf.js";
+import { I as z, h as V, W as F, X as H, u as S } from "./tensor_util-DfwaWayG.js";
+import { f as v } from "./backend_util-B1XRLuq9.js";
 const he = `
   if (isnan(a)) return a;
   if (isnan(b)) return b;
@@ -22,11 +22,11 @@ class b {
     `;
   }
 }
-function H(t, e) {
+function Y(t, e) {
   return ["x", "y", "z", "w", "u", "v"].slice(0, e).map((s) => `${t}.${s}`);
 }
 function Z(t, e) {
-  return e === 1 ? [t] : H(t, e);
+  return e === 1 ? [t] : Y(t, e);
 }
 function fe(t, e) {
   if (t === 1)
@@ -144,7 +144,7 @@ function j(t) {
   return s.runWebGLProgram(n, [r, u], "float32");
 }
 const Ne = {
-  kernelName: Y,
+  kernelName: H,
   backendName: "webgl",
   kernelFunc: j
 };
@@ -293,7 +293,7 @@ export {
   ye as d,
   M as e,
   T as f,
-  H as g,
+  Y as g,
   b as h,
   P as i,
   he as j,

package/dist/layers/BaseLayer.js

@@ -1,7 +1,7 @@
-import { w as p, e as o, F as g } from "../index-ZyQhjEPo.js";
-import { v } from "../variable-Bhn5bHYv.js";
-import { T as _ } from "../tensor-DdQUJZlz.js";
-class T {
+import { D as p, w as g, e as o, F as v } from "../index-D6Q1lPZO.js";
+import { v as _ } from "../variable-DzfrwYuP.js";
+import { T as V } from "../tensor-CzmOBsdf.js";
+class F {
   parent;
   config;
   _variables = /* @__PURE__ */ new Map();
@@ -70,7 +70,7 @@ class T {
   }
   saveWeights(t) {
     this._variables.forEach((r, e) => {
-      r && t.set(e, [r.clone()]);
+      r && t.set(e, [p(r)]);
     }), this.children.forEach((r) => {
       r.saveWeights(t);
     });
@@ -80,7 +80,7 @@ class T {
       const i = t.get(e)?.[0];
       if (!i)
         throw new Error(`Weights for ${e} not found`);
-      r ? r.assign(i) : this._variables.set(e, v(i, this._trainable));
+      r ? r.assign(i) : this._variables.set(e, _(i, this._trainable));
     }), this.children.forEach((r) => {
       r.loadWeights(t);
     });
@@ -98,7 +98,7 @@ class T {
   call(t, ...r) {
     this.build();
     const e = this.forward(t, ...r);
-    if (t.training && e instanceof _) {
+    if (t.training && e instanceof V) {
       const i = this.dropout(e);
       return i !== e && e.dispose(), i;
     } else
@@ -108,12 +108,12 @@ class T {
     return this.build(), this.checkpointingFn(t, ...r);
   }
   checkpointingFn(t, ...r) {
-    const e = this.trainableVariables, s = p((...a) => {
+    const e = this.trainableVariables, s = g((...a) => {
       const l = a[a.length - 1], n = a.slice(0, r.length), h = this.forward(t, ...n);
       return l(n), { value: h, gradFunc: (c, f) => {
         const u = o().state.activeTape;
         o().state.activeTape = [];
-        const b = g((...d) => this.forward(t, ...d.slice(0, n.length)))([...f, ...e], c);
+        const b = v((...d) => this.forward(t, ...d.slice(0, n.length)))([...f, ...e], c);
         return o().state.activeTape = u, b;
       } };
     })(...r, ...e);
@@ -125,5 +125,5 @@ class T {
   }
 }
 export {
-  T as default
+  F as default
 };

package/dist/layers/CausalSelfAttention.js

@@ -1,18 +1,18 @@
 import { attentionMask as P } from "../ops/attentionMask.js";
 import _ from "./BaseLayer.js";
-import { r as k } from "../rope-B5UUMsPi.js";
+import { r as k } from "../rope-s4W2XO9B.js";
 import { appendCache as b } from "../ops/appendCache.js";
-import { k as u, t as S } from "../index-ZyQhjEPo.js";
+import { k as u, t as S } from "../index-D6Q1lPZO.js";
 import { softmax16 as C } from "../ops/softmax16.js";
-import { b as R } from "../matMul16--R5hOwDG.js";
-import { p as V } from "../pack16-CFUqumar.js";
+import { b as R } from "../matMul16-fEAJ4smh.js";
+import { p as V } from "../pack16-CmVZs6af.js";
 import { transpose16 as J } from "../ops/transpose16.js";
 import { dot16 as L } from "../ops/dot16.js";
 import { reshape16 as N } from "../ops/reshape16.js";
 import { isPackedTensor as l } from "../utilities/packed.js";
 import { qkv as j } from "../ops/qkv.js";
-import { v as T } from "../variable-Bhn5bHYv.js";
-import { r as v, d as y } from "../dropout-C1pM3f11.js";
+import { v as T } from "../variable-DzfrwYuP.js";
+import { r as v, d as y } from "../dropout-B1x1kYMa.js";
 class X extends _ {
   divisor;
   index;

package/dist/layers/MLP.js

@@ -1,9 +1,9 @@
-import { t as p } from "../index-ZyQhjEPo.js";
+import { t as p } from "../index-D6Q1lPZO.js";
 import u from "./BaseLayer.js";
-import { c, b as M } from "../matMul16--R5hOwDG.js";
+import { c, b as M } from "../matMul16-fEAJ4smh.js";
 import { reshape16 as o } from "../ops/reshape16.js";
-import { v as d } from "../variable-Bhn5bHYv.js";
-import { r as h, d as b } from "../dropout-C1pM3f11.js";
+import { v as d } from "../variable-DzfrwYuP.js";
+import { r as h, d as b } from "../dropout-B1x1kYMa.js";
 class O extends u {
   index;
   hiddenUnits;

package/dist/layers/PositionEmbedding.js

@@ -1,9 +1,9 @@
-import { t as c, I as u, b as i } from "../index-ZyQhjEPo.js";
+import { t as c, I as u, b as i } from "../index-D6Q1lPZO.js";
 import f from "./BaseLayer.js";
-import { E as g, D as h } from "../random_width-DY6Kk2Dl.js";
-import { r as b } from "../exports_initializers-xuidcwI4.js";
-import { m as l } from "../mod-Gt1rMB4n.js";
-import { r as w } from "../range-BMS52eQi.js";
+import { E as g, D as h } from "../random_width-BVV9HveY.js";
+import { r as b } from "../exports_initializers-CZSUJoVE.js";
+import { m as l } from "../mod-DESSvHIU.js";
+import { r as w } from "../range-ZZZD60Fx.js";
 function E(t) {
   return new h(t);
 }

package/dist/layers/RMSNorm.js

@@ -1,8 +1,8 @@
-import { t as s } from "../index-ZyQhjEPo.js";
+import { t as s } from "../index-D6Q1lPZO.js";
 import e from "./BaseLayer.js";
 import { normRMS as a } from "../ops/normRMS.js";
-import { v as i } from "../variable-Bhn5bHYv.js";
-import { o as m } from "../ones-CAMiP4I2.js";
+import { v as i } from "../variable-DzfrwYuP.js";
+import { o as m } from "../ones-jU9jlQvM.js";
 class l extends e {
   GAMMA;
   constructor(r, t = "", o) {

package/dist/layers/RoPECache.js

@@ -1,7 +1,7 @@
-import { b as n, p as a, t as p, k as r } from "../index-ZyQhjEPo.js";
-import { r as c } from "../reciprocal-CTmshQ9J.js";
-import { c as l, s as f } from "../sin-BGfy2HZo.js";
-import { r as h } from "../range-BMS52eQi.js";
+import { b as n, p as a, t as p, k as r } from "../index-D6Q1lPZO.js";
+import { r as c } from "../reciprocal-CrYlsAGD.js";
+import { c as l, s as f } from "../sin-Djs4aQiu.js";
+import { r as h } from "../range-ZZZD60Fx.js";
 class g {
   constructor(s) {
     this.config = s;

package/dist/layers/TiedEmbedding.js

@@ -1,13 +1,13 @@
-import "../random_width-DY6Kk2Dl.js";
-import "../index-ZyQhjEPo.js";
-import { r as s } from "../exports_initializers-xuidcwI4.js";
+import "../random_width-BVV9HveY.js";
+import "../index-D6Q1lPZO.js";
+import { r as s } from "../exports_initializers-CZSUJoVE.js";
 import a from "./BaseLayer.js";
 import { dot16 as o } from "../ops/dot16.js";
 import { isPackedTensor as r } from "../utilities/packed.js";
-import { p as m } from "../pack16-CFUqumar.js";
+import { p as m } from "../pack16-CmVZs6af.js";
 import { transpose16 as d } from "../ops/transpose16.js";
-import { v as p } from "../variable-Bhn5bHYv.js";
-import { g as h } from "../gather-DykLGqmW.js";
+import { v as p } from "../variable-DzfrwYuP.js";
+import { g as h } from "../gather-CH9sdacz.js";
 class g extends a {
   vocabSize;
   embedDim;

package/dist/layers/TransformerBlock.js

@@ -2,7 +2,7 @@ import p from "./CausalSelfAttention.js";
 import m from "./MLP.js";
 import d from "./RMSNorm.js";
 import h from "./BaseLayer.js";
-import { k as n, t as u } from "../index-ZyQhjEPo.js";
+import { k as n, t as u } from "../index-D6Q1lPZO.js";
 import { add16 as l } from "../ops/add16.js";
 class g extends h {
   ln1;

package/dist/loader/loadTransformers.js

@@ -1,7 +1,7 @@
 import l from "../tokeniser/CharTokeniser.js";
 import c from "../tokeniser/bpe.js";
 import { load_safetensors as b } from "../utilities/safetensors.js";
-import { H as y } from "../index-ZyQhjEPo.js";
+import { H as y } from "../index-D6Q1lPZO.js";
 import { dummyPassAsync as u } from "../utilities/dummy.js";
 import _ from "../models/factory.js";
 async function L(e, a, r, t) {