@genai-fi/nanogpt 1.4.0 → 1.4.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -201,6 +201,8 @@ var m = class m {
201
201
  this._training?.training || this.setStatus("ready");
202
202
  }), this._training.on("error", (e) => {
203
203
  this.setStatus("error"), this.ee.emit("error", e);
204
+ }), this._training.on("mode", () => {
205
+ this.ee.emit("mode", this._model?.metaData.mode);
204
206
  });
205
207
  }
206
208
  return this._training;
@@ -41,6 +41,7 @@ interface TrainingEvents {
41
41
  pausing: (id: string) => void;
42
42
  cancelling: (id: string) => void;
43
43
  progress: (job: ITrainingJob) => void;
44
+ mode: () => void;
44
45
  }
45
46
  export default class Training {
46
47
  private ee;
@@ -152,8 +152,8 @@ var o = class {
152
152
  u.datasetId = o.datasetId;
153
153
  }
154
154
  if (!u.trainDataset) try {
155
- let t = await i(e, this._model, this._tokeniser, o, u.datasetId ?? a(s), c, s);
156
- u.trainDataset = t.trainDataset, u.validationDataset = t.validationDataset, u.totalTokens = t.totalTokens;
155
+ let t = this._model.metaData.mode, n = await i(e, this._model, this._tokeniser, o, u.datasetId ?? a(s), c, s);
156
+ t !== this._model.metaData.mode && this.ee.emit("mode"), u.trainDataset = n.trainDataset, u.validationDataset = n.validationDataset, u.totalTokens = n.totalTokens;
157
157
  } catch (e) {
158
158
  throw this.setState(u, "error", /* @__PURE__ */ Error("prepare_data_failed")), u.trainer.dispose(), this._jobs.delete(u.id), e;
159
159
  }
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@genai-fi/nanogpt",
3
- "version": "1.4.0",
3
+ "version": "1.4.1",
4
4
  "type": "module",
5
5
  "main": "dist/main.js",
6
6
  "types": "dist/main.d.ts",