@genai-fi/nanogpt 0.6.2 → 0.7.0

package/dist/{zeros-YJDE7oRb.js → zeros-DCPCdFGq.js}

@@ -1,4 +1,4 @@
-import { o as m, q as r, Z as l, E as c, _ as i, x as p, $ as u, g as x } from "./index-BAzbokzv.js";
+import { x as m, y as r, a0 as l, E as c, a1 as i, C as p, a2 as u, j as x } from "./index-C0dhsYom.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -16,9 +16,9 @@ import { o as m, q as r, Z as l, E as c, _ as i, x as p, $ as u, g as x } from "
  * =============================================================================
  */
 function f(a, e) {
-  const o = r(a, "real", "complex"), s = r(e, "imag", "complex");
-  l(o.shape, s.shape, `real and imag shapes, ${o.shape} and ${s.shape}, must match in call to tf.complex().`);
-  const n = { real: o, imag: s };
+  const s = r(a, "real", "complex"), o = r(e, "imag", "complex");
+  l(s.shape, o.shape, `real and imag shapes, ${s.shape} and ${o.shape}, must match in call to tf.complex().`);
+  const n = { real: s, imag: o };
   return c.runKernel(i, n);
 }
 const g = /* @__PURE__ */ m({ complex_: f });
@@ -40,11 +40,11 @@ const g = /* @__PURE__ */ m({ complex_: f });
  */
 function t(a, e = "float32") {
   if (p(a), e === "complex64") {
-    const s = t(a, "float32"), n = t(a, "float32");
-    return g(s, n);
+    const o = t(a, "float32"), n = t(a, "float32");
+    return g(o, n);
   }
-  const o = u(x(a), e);
-  return c.makeTensor(o, a, e);
+  const s = u(x(a), e);
+  return c.makeTensor(s, a, e);
 }
 export {
   g as c,

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@genai-fi/nanogpt",
-    "version": "0.6.2",
+    "version": "0.7.0",
     "type": "module",
     "main": "dist/main.js",
     "types": "dist/main.d.ts",
@@ -50,6 +50,7 @@
     "dependencies": {
         "@dsnp/parquetjs": "^1.8.7",
         "@tensorflow/tfjs": "^4.22.0",
+        "@tensorflow/tfjs-backend-webgpu": "^4.22.0",
         "eventemitter3": "^5.0.1",
         "jszip": "^3.10.1",
         "papaparse": "^5.5.3",