@genai-fi/nanogpt 0.7.1 → 0.7.2

package/dist/{register_all_kernels-DnbAyBXt.js → register_all_kernels-BwDSRN-f.js}

@@ -1,35 +1,35 @@
-import { B as oe, C as se, E as le, cl as ps, cm as hs, co as en, cp as fs, cq as ms, cs as gs, I as xs, cr as Cs, ct as bs, bp as To, l as Y, dd as tn, cz as sn, n as Do, z as nn, br as Eo, cD as Is, cE as ys, cF as $s, aq as Nc, cI as on, y as vc, a as an, k as Ne, dl as ks, cL as rn, cM as ws, X as It, bw as Fo, cQ as Ns, dg as cn, dj as ln, b as Po, bA as Oo, ds as un, cS as vs, dv as dn, dw as pn, al as xe, dx as S, aT as Sc, o as Rc, T as Ao, dy as Tc, cT as Dc, x as Ec, q as Fc, aU as Pc, c as Oc, ac as L, d2 as _o, d3 as Lo, e as Me, aO as Vt, d7 as Vo, aM as Wo, d4 as Ac, aL as Wt, aj as Ke, d8 as Mo, as as hn, an as _c, j as z, bg as Ge, ap as Lc, at as fn, au as mn, K as Vc, a2 as Wc, J as Bo, a1 as ne, cd as Go, cn as zo, bE as Uo, p as me, a7 as st, bF as Ho, bG as Xo, bH as Ko, ab as ct, bI as qo, cu as jo, cv as Yo, bL as Qo, bK as Zo, aP as Jo, dc as ea, bM as gn, cx as ta, aY as sa, Q as na, bN as oa, by as Te, c9 as aa, bO as ra, bP as ia, cy as ca, bQ as la, a8 as xn, bR as Cn, b6 as ua, bS as da, u as nt, H as Mc, bT as pa, bU as ha, bV as fa, cb as ma, cc as ga, de as xa, bo as Ss, bi as Ft, bl as bn, cA as Kn, dz as mt, dA as Ca, cB as qn, df as Bc, Y as Gc, aR as ba, cC as Ia, bW as In, aV as ya, az as $a, ao as yt, d9 as ka, da as wa, b7 as Na, ca as va, di as Sa, a4 as Ra, G as Ta, dk as Da, bX as yn, b0 as $n, bZ as kn, cH as Ea, U as zc, b_ as Fa, b$ as Pa, cJ as Oa, cK as Aa, dm as _a, c0 as La, aS as Va, bh as Wa, S as Ma, ad as Ba, b9 as Ga, bb as za, ba as Ua, c1 as Ha, cU as Xa, c2 as Ka, P as qa, a5 as ja, c3 as Ya, cN as wn, dB as Qa, dC as Za, dD as Ja, R as er, V as Nn, bc as tr, cO as sr, bd as nr, cP as or, c5 as ar, b8 as rr, b3 as vn, a3 as ir, dn as cr, aQ as lr, c6 as Sn, a9 as Rn, c7 as Tn, bY as Dn, c4 as ur, dE as dr, dF as pr, dp as hr, dq as fr, dr as mr, L as gr, cR as xr, av as En, dt as Cr, dG as br, dH as Ir, bJ as Fn, du as yr, N as $r, be as kr, ax as Pn, b4 as wr, c8 as Nr, f as vr, dI as Sr, dJ as jn, d5 as Yn, d6 as Uc, t as Hc, dK as Xc, dL as Kc, bu as qc, b5 as jc, bf as Yc, bm as Qc, W as Zc, bn as Jc, bv as el, b2 as tl, bj as sl, bk as nl, aI as Rr, bq as ol, ae as al, ak as rl, bs as il, dh as cl, aW as ll, aX as ul, aZ as dl, a_ as pl, af as hl, bt as fl, b1 as ml, bC as gl, a$ as xl, ay as Tr, bx as Cl, ar as bl, bz as Il, bB as yl } from "./index-UdZhlibC.js";
-import { e as Dr, s as Er, l as Fr, a as Pr, b as Or, m as Ar, c as _r, r as Gs, f as $l, g as kl, h as wl, i as Nl, j as vl, k as Sl, n as Rl, o as Tl, q as Dl, t as El, u as Fl, v as Pl, w as Ol, x as Al, y as _l, z as Ll, A as Vl, B as Wl, C as Ml, p as Bl, F as Gl, G as zl, H as Ul, I as Hl, J as Xl, K as Kl, L as ql } from "./random_width-DbSpgl4o.js";
-import { r as re } from "./reshape-BR0eoLYN.js";
-import { b as jl } from "./broadcast_to-Ba9h_8DO.js";
-import { c as os } from "./concat-CbXTetof.js";
-import { c as Yl, s as Ql } from "./sin-CiBxrDqX.js";
-import { w as Lr, o as Vr, n as ss, a as zs, r as Us, p as Wr, f as Zl, g as Jl, d as eu, b as tu, e as su, k as nu, m as ou, q as au, v as ru, x as iu, s as cu, c as lu, t as uu, u as du } from "./ops-aRTXR2Sr.js";
-import { m as Tt } from "./mat_mul-dwmZz69e.js";
-import { w as Mr, k as Ye, h as Ve, x as pu, z as hu, A as fu, y as mu, D as gu, v as xu, i as ot, j as lt, d as De, e as $t, g as Rs, S as Br, a as Gr, B as zr, C as Ur, p as Cu, q as bu, r as Iu, u as yu } from "./selu_util-Ce6pu9IM.js";
-import { e as $u, l as ku, a as wu } from "./log_sum_exp-BnmCkHWl.js";
-import { f as Nu } from "./dropout-DPfPgWWe.js";
-import { g as vu } from "./gather-Bbh8DHhM.js";
-import { c as Hs, z as Hr } from "./zeros-DnPT2nD4.js";
-import { s as as } from "./slice-BHbDHObE.js";
-import { m as Su } from "./max-ByjEGoFx.js";
-import { r as Ru } from "./reciprocal-CGB48wZB.js";
-import { s as Xs } from "./split-CRU0PjVV.js";
-import { s as Tu } from "./softmax-JMEIUo2J.js";
-import { s as Du } from "./stack-ikk2Y8_P.js";
-import { s as Eu } from "./sum-NLYbiDag.js";
-import { a as j, u as ue, c as ut, i as dt, b as Fu, d as Pt, t as Pe, e as kt, f as gt, g as Xr, r as Ot, h as Be, j as Pu, k as Ou, l as On, z as Au, m as An, n as Kr, o as _u, p as Lu, q as Vu, v as Wu, w as Mu, x as Bu, y as Gu, A as zu, B as Uu, C as Hu, D as ft, E as Xu, F as Ku, G as qr, H as qu, I as ju, J as Yu, K as Qu, L as Zu, M as Ju, N as ed, O as td, P as sd, Q as nd, R as od, S as ad, T as rd, U as id, V as cd, W as ld, X as ud, Y as dd, Z as pd, _ as hd, $ as fd, a0 as md, a1 as gd, a2 as xd, a3 as Cd, a4 as bd, a5 as Id, a6 as yd, a7 as $d, a8 as kd, a9 as wd, aa as Nd, ab as vd, ac as Sd, ad as Rd, ae as Td, af as Dd, ag as Ed, ah as Fd, ai as Pd } from "./shared-Czipaeb6.js";
-import { m as xt, A as Ts, B as Ds, C as Es, D as jr, F as Yr, H as Qr, G as je, I as Zr, J as Jr, K as ei, L as ti, M as si, z as ni, y as oi, x as ai, w as ri, u as ii, E as ci, T as li, U as Od, V as Ad, W as _d, X as Ld, Y as Vd, Z as Wd, N as ui, O as di, P as pi, Q as hi, c as fi, f as Md, S as Bd, _ as Gd } from "./backend_util-C9Ut8n0Q.js";
-import { g as ye, a as $e, b as Re, c as Ee, e as Oe, h as Fs } from "./axis_util-5DTW2tFV.js";
+import { B as oe, C as se, E as le, cl as ps, cm as hs, co as en, cp as fs, cq as ms, cs as gs, I as xs, cr as Cs, ct as bs, bp as To, l as Y, dd as tn, cz as sn, n as Do, z as nn, br as Eo, cD as Is, cE as ys, cF as $s, aq as Nc, cI as on, y as vc, a as an, k as Ne, dl as ks, cL as rn, cM as ws, X as It, bw as Fo, cQ as Ns, dg as cn, dj as ln, b as Po, bA as Oo, ds as un, cS as vs, dv as dn, dw as pn, al as xe, dx as S, aT as Sc, o as Rc, T as Ao, dy as Tc, cT as Dc, x as Ec, q as Fc, aU as Pc, c as Oc, ac as L, d2 as _o, d3 as Lo, e as Me, aO as Vt, d7 as Vo, aM as Wo, d4 as Ac, aL as Wt, aj as Ke, d8 as Mo, as as hn, an as _c, j as z, bg as Ge, ap as Lc, at as fn, au as mn, K as Vc, a3 as Wc, J as Bo, a1 as ne, cd as Go, cn as zo, bE as Uo, p as me, a7 as st, bF as Ho, bG as Xo, bH as Ko, ab as ct, bI as qo, cu as jo, cv as Yo, bL as Qo, bK as Zo, aP as Jo, dc as ea, bM as gn, cx as ta, aY as sa, Q as na, bN as oa, by as Te, c9 as aa, bO as ra, bP as ia, cy as ca, bQ as la, a8 as xn, bR as Cn, b6 as ua, bS as da, u as nt, H as Mc, bT as pa, bU as ha, bV as fa, cb as ma, cc as ga, de as xa, bo as Ss, bi as Ft, bl as bn, cA as Kn, dz as mt, dA as Ca, cB as qn, df as Bc, Y as Gc, aR as ba, cC as Ia, bW as In, aV as ya, az as $a, ao as yt, d9 as ka, da as wa, b7 as Na, ca as va, di as Sa, a5 as Ra, G as Ta, dk as Da, bX as yn, b0 as $n, bZ as kn, cH as Ea, U as zc, b_ as Fa, b$ as Pa, cJ as Oa, cK as Aa, dm as _a, c0 as La, aS as Va, bh as Wa, S as Ma, ad as Ba, b9 as Ga, bb as za, ba as Ua, c1 as Ha, cU as Xa, c2 as Ka, P as qa, a2 as ja, c3 as Ya, cN as wn, dB as Qa, dC as Za, dD as Ja, R as er, V as Nn, bc as tr, cO as sr, bd as nr, cP as or, c5 as ar, b8 as rr, b3 as vn, a4 as ir, dn as cr, aQ as lr, c6 as Sn, a9 as Rn, c7 as Tn, bY as Dn, c4 as ur, dE as dr, dF as pr, dp as hr, dq as fr, dr as mr, L as gr, cR as xr, av as En, dt as Cr, dG as br, dH as Ir, bJ as Fn, du as yr, N as $r, be as kr, ax as Pn, b4 as wr, c8 as Nr, f as vr, dI as Sr, dJ as jn, d5 as Yn, d6 as Uc, t as Hc, dK as Xc, dL as Kc, bu as qc, b5 as jc, bf as Yc, bm as Qc, W as Zc, bn as Jc, bv as el, b2 as tl, bj as sl, bk as nl, aI as Rr, bq as ol, ae as al, ak as rl, bs as il, dh as cl, aW as ll, aX as ul, aZ as dl, a_ as pl, af as hl, bt as fl, b1 as ml, bC as gl, a$ as xl, ay as Tr, bx as Cl, ar as bl, bz as Il, bB as yl } from "./index-BoWRt-10.js";
+import { e as Dr, s as Er, l as Fr, a as Pr, b as Or, m as Ar, c as _r, r as Gs, f as $l, g as kl, h as wl, i as Nl, j as vl, k as Sl, n as Rl, o as Tl, q as Dl, t as El, u as Fl, v as Pl, w as Ol, x as Al, y as _l, z as Ll, A as Vl, B as Wl, C as Ml, p as Bl, F as Gl, G as zl, H as Ul, I as Hl, J as Xl, K as Kl, L as ql } from "./random_width-sZORGo5k.js";
+import { r as re } from "./reshape-CdBq1WJ6.js";
+import { b as jl } from "./broadcast_to-CdbwV-Dj.js";
+import { c as os } from "./concat-CsxrgovM.js";
+import { c as Yl, s as Ql } from "./sin-BeA3tsEd.js";
+import { w as Lr, o as Vr, n as ss, a as zs, r as Us, p as Wr, f as Zl, g as Jl, d as eu, b as tu, e as su, k as nu, m as ou, q as au, v as ru, x as iu, s as cu, c as lu, t as uu, u as du } from "./ops-BFGCx8Ri.js";
+import { m as Tt } from "./mat_mul-8m8pfdcx.js";
+import { w as Mr, k as Ye, h as Ve, x as pu, z as hu, A as fu, y as mu, D as gu, v as xu, i as ot, j as lt, d as De, e as $t, g as Rs, S as Br, a as Gr, B as zr, C as Ur, p as Cu, q as bu, r as Iu, u as yu } from "./selu_util-BJEXVvjX.js";
+import { e as $u, l as ku, a as wu } from "./log_sum_exp-DbjkV734.js";
+import { f as Nu } from "./dropout-DYs5QFGQ.js";
+import { g as vu } from "./gather-CMMy2KEG.js";
+import { c as Hs, z as Hr } from "./zeros--BdLQ3oG.js";
+import { s as as } from "./slice-BiOsknYS.js";
+import { m as Su } from "./max-Ddnnb5xe.js";
+import { r as Ru } from "./reciprocal-BvGAyKyu.js";
+import { s as Xs } from "./split-B-dikLRw.js";
+import { s as Tu } from "./softmax-Bv_6lyMX.js";
+import { s as Du } from "./stack-B17UN2nn.js";
+import { s as Eu } from "./sum-66ew2byf.js";
+import { a as j, u as ue, c as ut, i as dt, b as Fu, d as Pt, t as Pe, e as kt, f as gt, g as Xr, r as Ot, h as Be, j as Pu, k as Ou, l as On, z as Au, m as An, n as Kr, o as _u, p as Lu, q as Vu, v as Wu, w as Mu, x as Bu, y as Gu, A as zu, B as Uu, C as Hu, D as ft, E as Xu, F as Ku, G as qr, H as qu, I as ju, J as Yu, K as Qu, L as Zu, M as Ju, N as ed, O as td, P as sd, Q as nd, R as od, S as ad, T as rd, U as id, V as cd, W as ld, X as ud, Y as dd, Z as pd, _ as hd, $ as fd, a0 as md, a1 as gd, a2 as xd, a3 as Cd, a4 as bd, a5 as Id, a6 as yd, a7 as $d, a8 as kd, a9 as wd, aa as Nd, ab as vd, ac as Sd, ad as Rd, ae as Td, af as Dd, ag as Ed, ah as Fd, ai as Pd } from "./shared-B8ztnyEk.js";
+import { m as xt, A as Ts, B as Ds, C as Es, D as jr, F as Yr, H as Qr, G as je, I as Zr, J as Jr, K as ei, L as ti, M as si, z as ni, y as oi, x as ai, w as ri, u as ii, E as ci, T as li, U as Od, V as Ad, W as _d, X as Ld, Y as Vd, Z as Wd, N as ui, O as di, P as pi, Q as hi, c as fi, f as Md, S as Bd, _ as Gd } from "./backend_util-TE7aTPhZ.js";
+import { g as ye, a as $e, b as Re, c as Ee, e as Oe, h as Fs } from "./axis_util-BzbKo31C.js";
 import { s as zd } from "./index-C4L8Cm77.js";
 import { n as mi, b as gi, a as xi } from "./non_max_suppression_impl-CsEgBuMA.js";
-import { c as wt } from "./scatter_nd_util-OjyAxku2.js";
-import { h as Qn, j as Ud, k as Hd, l as Xd, m as Kd, n as qd, o as jd, P as _n, p as ze, u as We, q as Ci, c as bi, T as _e, E as Ii, g as yi, a as $i, r as Yd, s as Qd, t as Z, v as Mt, w as Zd, x as Zn, y as Jd, z as ep, A as Bt, B as tp, C as sp, D as Ks, F as jt, G as Yt, H as np, I as op, J as Jn, K as ap, L as rp, M as Vs, N as ip, O as cp, Q as lp, R as Qt, S as Ws, U as up, f as Ce, V as we, W as Zt, X as Jt, Y as dp, d as eo, e as to, i as ki, Z as pp, _ as hp, $ as fp, a0 as mp, a1 as gp, a2 as xp, a3 as Gt } from "./gpgpu_math-DffelNS-.js";
-import { s as wi, a as Cp, t as Ni, b as bp, c as Ip, d as vi, e as yp, n as $p, f as kp, g as wp, h as Np, i as vp, j as Sp, k as Rp, l as Tp, o as Dp, p as Ep, q as Fp, r as Pp, u as Op, v as Ap, w as _p, x as Lp, y as Vp, z as Wp, A as Mp, B as Bp, C as Gp, D as zp, E as Up, F as Hp, G as Xp, H as Kp, I as qp, J as jp, K as Yp, L as Qp, M as Si, N as Zp, O as Jp, P as eh, Q as th, R as sh, S as nh, T as oh, U as ah, V as rh, W as ih } from "./shared-DS5waSIY.js";
-import { a as Se, c as ch, U as it, d as tt, e as qe, A as so, f as Nt, B as Ln, h as Vn, m as At, u as ae, C as Ue, b as ke, i as Le, j as Wn, k as pt, l as vt, n as lh, o as uh, p as dh, q as ph } from "./kernel_funcs_utils-CXDy3EN7.js";
-import { R as hh, r as H, a as fh } from "./Reshape-CEsEp0AI.js";
-import { M as Ri } from "./mulmat_packed_gpu-IGPBp6h9.js";
-import { t as Ti, s as Mn, a as zt, m as mh, r as gh, b as xh, c as Ch, d as bh } from "./RealDiv-CVYNbZxu.js";
+import { c as wt } from "./scatter_nd_util-DUstGbU1.js";
+import { h as Qn, j as Ud, k as Hd, l as Xd, m as Kd, n as qd, o as jd, P as _n, p as ze, u as We, q as Ci, c as bi, T as _e, E as Ii, g as yi, a as $i, r as Yd, s as Qd, t as Z, v as Mt, w as Zd, x as Zn, y as Jd, z as ep, A as Bt, B as tp, C as sp, D as Ks, F as jt, G as Yt, H as np, I as op, J as Jn, K as ap, L as rp, M as Vs, N as ip, O as cp, Q as lp, R as Qt, S as Ws, U as up, f as Ce, V as we, W as Zt, X as Jt, Y as dp, d as eo, e as to, i as ki, Z as pp, _ as hp, $ as fp, a0 as mp, a1 as gp, a2 as xp, a3 as Gt } from "./gpgpu_math-DGNLNL4I.js";
+import { s as wi, a as Cp, t as Ni, b as bp, c as Ip, d as vi, e as yp, n as $p, f as kp, g as wp, h as Np, i as vp, j as Sp, k as Rp, l as Tp, o as Dp, p as Ep, q as Fp, r as Pp, u as Op, v as Ap, w as _p, x as Lp, y as Vp, z as Wp, A as Mp, B as Bp, C as Gp, D as zp, E as Up, F as Hp, G as Xp, H as Kp, I as qp, J as jp, K as Yp, L as Qp, M as Si, N as Zp, O as Jp, P as eh, Q as th, R as sh, S as nh, T as oh, U as ah, V as rh, W as ih } from "./shared-wS99K7_n.js";
+import { a as Se, c as ch, U as it, d as tt, e as qe, A as so, f as Nt, B as Ln, h as Vn, m as At, u as ae, C as Ue, b as ke, i as Le, j as Wn, k as pt, l as vt, n as lh, o as uh, p as dh, q as ph } from "./kernel_funcs_utils-BYKWV8Aa.js";
+import { R as hh, r as H, a as fh } from "./Reshape-DH5srBP0.js";
+import { M as Ri } from "./mulmat_packed_gpu-VSekgsNv.js";
+import { t as Ti, s as Mn, a as zt, m as mh, r as gh, b as xh, c as Ch, d as bh } from "./RealDiv-Dy0p8Bvo.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{reshape-BR0eoLYN.js → reshape-CdBq1WJ6.js}

@@ -1,4 +1,4 @@
-import { B as o, C as t, E as a, K as p } from "./index-UdZhlibC.js";
+import { B as o, C as t, E as a, K as p } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{scatter_nd_util-OjyAxku2.js → scatter_nd_util-DUstGbU1.js}

@@ -1,4 +1,4 @@
-import { j as p, a1 as w } from "./index-UdZhlibC.js";
+import { j as p, a1 as w } from "./index-BoWRt-10.js";
 function k(o, t, r) {
   const n = t.rank > 1 ? t.shape[t.rank - 1] : 1, e = t.rank > 1 ? t.rank - 1 : 1, h = `Must have updates.shape = indices.shape[:batchDim] + shape[sliceDim:], got updates.shape: ${r.shape}, indices.shape: ${t.shape}, shape: ${o}, sliceDim: ${n}, and batchDim: ${e}.`;
   if (r.rank < e)

package/dist/{selu_util-Ce6pu9IM.js → selu_util-BJEXVvjX.js}

@@ -1,6 +1,6 @@
-import { l as N, aq as H, B as _, C as S, E as L, ar as te, as as ne, an as se, ap as re, at as ie, au as oe, av as le, a as ue, aw as fe, ax as O } from "./index-UdZhlibC.js";
-import { r as ce } from "./reshape-BR0eoLYN.js";
-import { s as ae } from "./sum-NLYbiDag.js";
+import { l as N, aq as H, B as _, C as S, E as L, ar as te, as as ne, an as se, ap as re, at as ie, au as oe, av as le, a as ue, aw as fe, ax as O } from "./index-BoWRt-10.js";
+import { r as ce } from "./reshape-CdBq1WJ6.js";
+import { s as ae } from "./sum-66ew2byf.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{shared-Czipaeb6.js → shared-B8ztnyEk.js}

@@ -1,9 +1,9 @@
-import { l as Lt, bf as Ce, j as V, k as Pt, a1 as H, bg as et, aa as ht, bh as dt, bi as at, a6 as De, a7 as st, am as We, b2 as $e, bj as Ue, bk as ze, bl as Be, bm as qt, al as K, bn as At, bo as W, bp as _t, bq as Vt, ae as Ct, br as Dt, ak as Wt, bs as $t, aW as Ut, aX as zt, aZ as Bt, a_ as jt, af as Gt, bt as Zt, b1 as Kt, bu as Ht, a$ as je, ao as Ge, bv as Xt, b5 as Ze, bw as Ke, p as Jt, u as He, t as Xe, bx as Qt, by as It, ar as Yt, W as Je, bz as te, bA as ee, bB as ne, aM as pt, bC as se, bD as ot } from "./index-UdZhlibC.js";
-import { f as it, m as kt, g as Qe, a as Ye, R as tn, v as en, d as nn, e as sn, h as on, i as rn, j as an, k as ln, l as cn, n as un, o as hn, p as fn, q as Nt, r as dn, s as gn, t as mn } from "./backend_util-C9Ut8n0Q.js";
-import { g as pn, a as wn, e as In, c as bn } from "./axis_util-5DTW2tFV.js";
-import { b as xn } from "./broadcast_to-Ba9h_8DO.js";
-import { r as En } from "./reshape-BR0eoLYN.js";
-import { p as Fn, q as yn, r as kn, u as Nn } from "./selu_util-Ce6pu9IM.js";
+import { l as Lt, bf as Ce, j as V, k as Pt, a1 as H, bg as et, aa as ht, bh as dt, bi as at, a6 as De, a7 as st, am as We, b2 as $e, bj as Ue, bk as ze, bl as Be, bm as qt, al as K, bn as At, bo as W, bp as _t, bq as Vt, ae as Ct, br as Dt, ak as Wt, bs as $t, aW as Ut, aX as zt, aZ as Bt, a_ as jt, af as Gt, bt as Zt, b1 as Kt, bu as Ht, a$ as je, ao as Ge, bv as Xt, b5 as Ze, bw as Ke, p as Jt, u as He, t as Xe, bx as Qt, by as It, ar as Yt, W as Je, bz as te, bA as ee, bB as ne, aM as pt, bC as se, bD as ot } from "./index-BoWRt-10.js";
+import { f as it, m as kt, g as Qe, a as Ye, R as tn, v as en, d as nn, e as sn, h as on, i as rn, j as an, k as ln, l as cn, n as un, o as hn, p as fn, q as Nt, r as dn, s as gn, t as mn } from "./backend_util-TE7aTPhZ.js";
+import { g as pn, a as wn, e as In, c as bn } from "./axis_util-BzbKo31C.js";
+import { b as xn } from "./broadcast_to-CdbwV-Dj.js";
+import { r as En } from "./reshape-CdBq1WJ6.js";
+import { p as Fn, q as yn, r as kn, u as Nn } from "./selu_util-BJEXVvjX.js";
 import { g as vn } from "./_commonjsHelpers-ByX85dGu.js";
 function Sn(e, t) {
   for (var n = 0; n < t.length; n++) {

package/dist/{shared-DS5waSIY.js → shared-wS99K7_n.js}

@@ -1,4 +1,4 @@
-import { s as l } from "./shared-Czipaeb6.js";
+import { s as l } from "./shared-B8ztnyEk.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{sin-CiBxrDqX.js → sin-BeA3tsEd.js}

@@ -1,4 +1,4 @@
-import { B as o, C as t, E as c, a8 as a, a9 as e } from "./index-UdZhlibC.js";
+import { B as o, C as t, E as c, a8 as a, a9 as e } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{slice-BHbDHObE.js → slice-BiOsknYS.js}

@@ -1,4 +1,4 @@
-import { B as c, C as e, E as a, W as l } from "./index-UdZhlibC.js";
+import { B as c, C as e, E as a, W as l } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{softmax-JMEIUo2J.js → softmax-Bv_6lyMX.js}

@@ -1,4 +1,4 @@
-import { B as r, C as f, E as e, S as i } from "./index-UdZhlibC.js";
+import { B as r, C as f, E as e, S as i } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{split-CRU0PjVV.js → split-B-dikLRw.js}

@@ -1,4 +1,4 @@
-import { B as p, C as i, E as a, L as c } from "./index-UdZhlibC.js";
+import { B as p, C as i, E as a, L as c } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{stack-ikk2Y8_P.js → stack-B17UN2nn.js}

@@ -1,4 +1,4 @@
-import { B as e, D as c, l as n, E as k, P as i } from "./index-UdZhlibC.js";
+import { B as e, D as c, l as n, E as k, P as i } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{sum-NLYbiDag.js → sum-66ew2byf.js}

@@ -1,4 +1,4 @@
-import { B as e, C as u, X as c, E as l, Y as m } from "./index-UdZhlibC.js";
+import { B as e, C as u, X as c, E as l, Y as m } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{tensor-Do9PKbIE.js → tensor-JwS7ZYY6.js}

@@ -1,4 +1,4 @@
-import { Z as t, _ as a } from "./index-UdZhlibC.js";
+import { Z as t, _ as a } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{tensor2d-CWHxHpLh.js → tensor2d-wxPAnDQy.js}

@@ -1,4 +1,4 @@
-import { $ as t, Z as s, _ as a } from "./index-UdZhlibC.js";
+import { $ as t, Z as s, _ as a } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/training/Adam.js

@@ -1,7 +1,7 @@
 import { adamAdjust as b } from "../ops/adamAdjust.js";
 import { adamMoments as d } from "../ops/adamMoments.js";
-import { O as g, e as h, t as o, d as B } from "../index-UdZhlibC.js";
-import { z as M } from "../zeros-DnPT2nD4.js";
+import { O as g, e as h, t as o, d as B } from "../index-BoWRt-10.js";
+import { z as M } from "../zeros--BdLQ3oG.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/training/AdamExt.js

@@ -1,4 +1,4 @@
-import { a as r, b as c, c as h, e as o } from "../index-UdZhlibC.js";
+import { a as r, b as c, c as h, e as o } from "../index-BoWRt-10.js";
 import { AdamOptimizer as g } from "./Adam.js";
 class y extends g {
   constructor(t, e, s, i, a) {

package/dist/training/DatasetBuilder.js

@@ -1,5 +1,5 @@
-import { t as u } from "../index-UdZhlibC.js";
-import { d as z, i as f } from "../dataset-U3PrjwgU.js";
+import { t as g } from "../index-BoWRt-10.js";
+import { d as u, i as d } from "../dataset-CtdBYwjo.js";
 import "../index-Tf7vU29b.js";
 /**
  * @license
@@ -18,57 +18,60 @@ import "../index-Tf7vU29b.js";
  *
  * =============================================================================
  */
-function S(c) {
-  return z(async () => {
-    const t = await c();
-    return f(() => t.next());
+function z(r) {
+  return u(async () => {
+    const t = await r();
+    return d(() => t.next());
   });
 }
-const p = 8;
-async function y(c, t) {
-  const s = await Promise.all(c.map((n) => t.encode(n))), i = t.eosToken >= 0;
-  return s.map((n) => i ? [...n, t.eosToken] : n).flat();
+const S = 8;
+async function y(r, t) {
+  const s = await Promise.all(r.map((e) => t.encode(e))), o = t.eosToken >= 0, a = s.map((e) => o ? [...e, t.eosToken] : e).flat();
+  for (const e of a)
+    if (e < 0 || e >= t.vocabSize)
+      throw new Error(`Invalid token index ${e} found in tokenised data`);
+  return a;
 }
 class w {
   tokenizer;
   blockSize;
   pageSize;
   constructor(t, s = 128) {
-    this.tokenizer = t, this.blockSize = s, this.pageSize = s * p;
+    this.tokenizer = t, this.blockSize = s, this.pageSize = s * S;
   }
   // Create dataset from text files
-  async createTextDataset(t, s = 32, i, r) {
+  async createTextDataset(t, s = 32, o, a) {
     if (t.length < this.blockSize + 1)
       throw new Error(`Not enough tokens (${t.length}) for block size ${this.blockSize}`);
-    if (i && i.size > t.length / this.pageSize / 2)
+    if (o && o.size > t.length / this.pageSize / 2)
       throw new Error("Too many masked pages - would leave insufficient training data");
-    const n = (function* () {
-      if (i && r) {
-        const e = Array.from(i);
+    const e = (function* () {
+      if (o && a) {
+        const i = Array.from(o);
         for (; ; ) {
-          const a = Math.floor(Math.random() * e.length), l = Math.floor(Math.random() * this.pageSize), o = e[a] * this.pageSize + l;
-          if (o + this.blockSize + 1 > t.length)
+          const c = Math.floor(Math.random() * i.length), l = Math.floor(Math.random() * this.pageSize), n = i[c] * this.pageSize + l;
+          if (n + this.blockSize + 1 > t.length)
             continue;
-          const h = t.slice(o, o + this.blockSize), g = t.slice(o + 1, o + this.blockSize + 1);
-          yield { xs: h, ys: g };
+          const h = t.slice(n, n + this.blockSize), f = t.slice(n + 1, n + this.blockSize + 1);
+          yield { xs: h, ys: f };
         }
       } else
         for (; ; ) {
-          const e = Math.floor(Math.random() * (t.length - this.blockSize - 1));
-          if (i) {
-            const o = Math.floor(e / this.pageSize), h = i.has(o);
-            if (h && !r || !h && r)
+          const i = Math.floor(Math.random() * (t.length - this.blockSize - 1));
+          if (o) {
+            const n = Math.floor(i / this.pageSize), h = o.has(n);
+            if (h && !a || !h && a)
               continue;
           }
-          const a = t.slice(e, e + this.blockSize), l = t.slice(e + 1, e + this.blockSize + 1);
-          yield { xs: a, ys: l };
+          const c = t.slice(i, i + this.blockSize), l = t.slice(i + 1, i + this.blockSize + 1);
+          yield { xs: c, ys: l };
         }
     }).bind(this);
-    return S(n).batch(s).map((e) => {
-      const a = e;
-      return u(() => ({
-        xs: a.xs.cast("int32"),
-        ys: a.ys.cast("int32")
+    return z(e).batch(s).map((i) => {
+      const c = i;
+      return g(() => ({
+        xs: c.xs.cast("int32"),
+        ys: c.ys.cast("int32")
         // this.tf.oneHot(batchData.ys.cast('int32'), this.tokenizer.vocabSize),
       }));
     }).prefetch(2);
@@ -76,6 +79,6 @@ class w {
 }
 export {
   w as DatasetBuilder,
-  p as PAGE_FACTOR,
+  S as PAGE_FACTOR,
   y as flattenTokens
 };

package/dist/training/FullTrainer.js

@@ -1,7 +1,7 @@
 import { generateText as w } from "../utilities/generate.js";
 import T from "./Trainer.js";
 import L from "./Evaluator.js";
-import { d as h } from "../index-UdZhlibC.js";
+import { d as h } from "../index-BoWRt-10.js";
 import x from "../utilities/profile.js";
 const y = {
   desiredLoss: 0.01,

package/dist/training/Trainer.js

@@ -1,7 +1,7 @@
 import { DatasetBuilder as h, flattenTokens as p, PAGE_FACTOR as g } from "./DatasetBuilder.js";
 import u from "./AdamExt.js";
-import { t as f, v as y, d as c } from "../index-UdZhlibC.js";
-import { z as m } from "../zeros-DnPT2nD4.js";
+import { t as f, v as y, d as c } from "../index-BoWRt-10.js";
+import { z as m } from "../zeros--BdLQ3oG.js";
 class x {
   constructor(t, e, a = 1e-3) {
     this.tokenizer = e, this.model = t, this.learningRate = a, this.resetOptimizer(), this.datasetBuilder = new h(e, t.config.gpt.blockSize);

package/dist/training/sparseCrossEntropy.js

@@ -1,9 +1,9 @@
 import { gatherSub as x } from "../ops/gatherSub.js";
 import { scatterSub as L } from "../ops/scatterSub.js";
-import { y, t as u, z as C, c as E } from "../index-UdZhlibC.js";
-import { s as G } from "../softmax-JMEIUo2J.js";
-import { m as z } from "../max-ByjEGoFx.js";
-import { l as v } from "../log_sum_exp-BnmCkHWl.js";
+import { y, t as u, z as C, c as E } from "../index-BoWRt-10.js";
+import { s as G } from "../softmax-Bv_6lyMX.js";
+import { m as z } from "../max-Ddnnb5xe.js";
+import { l as v } from "../log_sum_exp-DbjkV734.js";
 function k(t, s) {
   return u(() => {
     const n = t.shape[t.shape.length - 1], c = t.shape.slice(0, -1).reduce((o, e) => o * e, 1), h = t.shape.length > 2 ? t.reshape([c, n]) : t, p = s.shape.length > 1 ? s.reshape([c]).cast("int32") : s.cast("int32"), r = z(h, -1, !0), a = E(h, r), m = v(a, -1);

package/dist/utilities/dummy.js

@@ -1,5 +1,5 @@
-import { m as y, v as P, e as S } from "../index-UdZhlibC.js";
-import { z as i } from "../zeros-DnPT2nD4.js";
+import { m as y, v as P, e as S } from "../index-BoWRt-10.js";
+import { z as i } from "../zeros--BdLQ3oG.js";
 async function w(s) {
   const t = i([1, s.config.gpt.blockSize], "int32"), [e, n] = s.forward({ training: !1 }, t);
   await e.data(), e.dispose(), n && n.dispose(), t.dispose();

package/dist/utilities/generate.js

@@ -1,6 +1,6 @@
-import "../index-UdZhlibC.js";
-import { t as m } from "../tensor2d-CWHxHpLh.js";
-import { c as u } from "../concat-CbXTetof.js";
+import "../index-BoWRt-10.js";
+import { t as m } from "../tensor2d-wxPAnDQy.js";
+import { c as u } from "../concat-CsxrgovM.js";
 async function v(o, r, a, c, f) {
   if (c <= 0)
     throw new Error("Length must be a positive integer");

package/dist/utilities/multinomialCPU.js

@@ -1,5 +1,5 @@
-import "../index-UdZhlibC.js";
-import { t as e } from "../tensor2d-CWHxHpLh.js";
+import "../index-BoWRt-10.js";
+import { t as e } from "../tensor2d-wxPAnDQy.js";
 function l(n) {
   let r = 0;
   const i = Math.random();

package/dist/utilities/performance.js

@@ -1,4 +1,4 @@
-import { t as s } from "../index-UdZhlibC.js";
+import { t as s } from "../index-BoWRt-10.js";
 async function f(e, o = 10, r = !1) {
   for (let t = 0; t < 100; t++) {
     const a = r ? await e() : s(e);

package/dist/utilities/profile.js

@@ -1,4 +1,4 @@
-import { m as a } from "../index-UdZhlibC.js";
+import { m as a } from "../index-BoWRt-10.js";
 const s = 1024 * 1024;
 class l {
   log = /* @__PURE__ */ new Map();

package/dist/utilities/safetensors.js

@@ -1,5 +1,5 @@
-import "../index-UdZhlibC.js";
-import { t as y } from "../tensor-Do9PKbIE.js";
+import "../index-BoWRt-10.js";
+import { t as y } from "../tensor-JwS7ZYY6.js";
 function l(t) {
   if (t === "float32") return "F32";
   if (t === "int32") return "I32";

package/dist/utilities/weights.js

@@ -1,5 +1,5 @@
-import "../index-UdZhlibC.js";
-import { t as p } from "../tensor-Do9PKbIE.js";
+import "../index-BoWRt-10.js";
+import { t as p } from "../tensor-JwS7ZYY6.js";
 function h(n) {
   const e = n.reduce((s, o) => s + o.length, 0), a = new Float32Array(e);
   let t = 0;

package/dist/{variable-BTBkayv_.js → variable-BuddVFLa.js}

@@ -1,4 +1,4 @@
-import { E as i } from "./index-UdZhlibC.js";
+import { E as i } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{webgpu_program-WaoMq-WD.js → webgpu_program-PFzf1hAQ.js}

@@ -1,4 +1,4 @@
-import { aa as k, ab as z, ac as E, a1 as j, l as A } from "./index-UdZhlibC.js";
+import { aa as k, ab as z, ac as E, a1 as j, l as A } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2019 Google LLC. All Rights Reserved.

package/dist/{webgpu_util-DhSeP4b6.js → webgpu_util-D____QpY.js}

@@ -1,4 +1,4 @@
-import { l as u } from "./index-UdZhlibC.js";
+import { l as u } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2019 Google LLC. All Rights Reserved.

package/dist/{zeros-DnPT2nD4.js → zeros--BdLQ3oG.js}

@@ -1,4 +1,4 @@
-import { B as m, C as r, a5 as l, E as c, a6 as i, F as p, a7 as u, j as f } from "./index-UdZhlibC.js";
+import { B as m, C as r, a2 as l, E as c, a6 as i, F as p, a7 as u, j as f } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@genai-fi/nanogpt",
-    "version": "0.7.1",
+    "version": "0.7.2",
     "type": "module",
     "main": "dist/main.js",
     "types": "dist/main.d.ts",