@genai-fi/nanogpt 0.5.5 → 0.5.6

package/dist/TeachableLLM.js

@@ -91,19 +91,19 @@ class a {
   }
   static loadModel(t) {
     const e = new a();
-    return f(t).then(({ model: i, tokeniser: o }) => {
-      e._model = i, e._tokeniser = o, e._config = i.config, e.setStatus("warmup"), m(i).then(() => {
+    return f(t).then(({ model: i, tokeniser: s }) => {
+      e._model = i, e._tokeniser = s, e._config = i.config, e.setStatus("warmup"), m(i).then(() => {
         e.setStatus("ready"), e.ee.emit("loaded");
-      }).catch((s) => {
-        e.setStatus("error"), e.ee.emit("error", s);
+      }).catch((o) => {
+        e.setStatus("error"), e.ee.emit("error", o);
       });
     }).catch((i) => {
       e.setStatus("error"), e.ee.emit("error", i);
     }), e;
   }
   static create(t, e = {}) {
-    const i = { ...h, ...e }, o = t === "char" ? new c(i.vocabSize) : new g(i.vocabSize), s = new l(i), r = new a(o, s);
-    return r.setStatus("warmup"), m(s).then(() => {
+    const i = { ...h, ...e }, s = t === "char" ? new c(i.vocabSize) : new g(i.vocabSize), o = new l(i), r = new a(s, o);
+    return r.setStatus("warmup"), m(o).then(() => {
       r.tokeniser.trained ? (r.setStatus("ready"), r.ee.emit("loaded")) : (r.setStatus("awaitingTokens"), r.ee.emit("loaded"), r.tokeniser.once("trainStatus", (n) => {
         n === "trained" && r.setStatus("ready");
       }));
@@ -134,10 +134,10 @@ class a {
     if (!this._model || !this._tokeniser)
       throw new Error("Model or tokeniser is not initialized.");
     const t = new p(this._model, this._tokeniser);
-    return t.on("start", () => this.setStatus("training")), t.on("stop", () => this.setStatus("ready")), t.on("log", async (e) => {
-      const i = this.ee.listeners("trainStep");
-      for (const o of i)
-        await o(e);
+    return t.on("start", () => this.setStatus("training")), t.on("stop", () => this.setStatus("ready")), t.on("log", async (e, i) => {
+      const s = this.ee.listeners("trainStep");
+      for (const o of s)
+        await o(e, i);
     }), t;
   }
   train(t, e) {

package/dist/main.d.ts

@@ -8,5 +8,6 @@ export type { ITrainerOptions } from './Trainer';
 export type { IGenerateOptions } from './Generator';
 export type { TrainingLogEntry } from './NanoGPTModel';
 export type { ITokeniser } from './tokeniser/type';
+export type { TrainingProgress } from './training/Trainer';
 export type { GPTConfig } from './config';
 export { estimateParameterCount, estimateMemoryUsage, estimateTrainingMemoryUsage, estimateResources, validateConfig, } from './utilities/parameters';

package/package.json

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