@genai-fi/nanogpt 0.21.0 → 0.23.0

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.
Files changed (55) hide show
  1. package/dist/{DatasetBuilder-B9UGNQT5.js → DatasetBuilder-C0iJT29K.js} +122 -93
  2. package/dist/Generator.js +2 -2
  3. package/dist/TeachableLLM.d.ts +2 -1
  4. package/dist/TeachableLLM.js +1 -1
  5. package/dist/Trainer-DBsyWJ4s.js +228 -0
  6. package/dist/Trainer.d.ts +2 -1
  7. package/dist/Trainer.js +1 -1
  8. package/dist/data/stream.d.ts +24 -0
  9. package/dist/data/stream.js +2 -0
  10. package/dist/data/textLoader.d.ts +2 -2
  11. package/dist/data/textLoader.js +28 -67
  12. package/dist/{BaseTokeniser-C9TSv4th.js → eventemitter3-D_qV3Lof.js} +2 -132
  13. package/dist/loader/load.js +2 -2
  14. package/dist/loader/loadHF.js +1 -1
  15. package/dist/loader/loadTransformers.js +2 -2
  16. package/dist/loader/newZipLoad.js +1 -1
  17. package/dist/loader/oldZipLoad.js +1 -1
  18. package/dist/loader/save.js +1 -1
  19. package/dist/{main-DdsoTJAW.js → main-BSaDGH7I.js} +2686 -2912
  20. package/dist/main.d.ts +4 -0
  21. package/dist/main.js +20 -16
  22. package/dist/models/NanoGPTV1.js +1 -1
  23. package/dist/models/NanoGPTV2.js +1 -1
  24. package/dist/models/factory.js +1 -1
  25. package/dist/models/model.js +1 -1
  26. package/dist/stream-BjdpSNqB.js +2296 -0
  27. package/dist/tokeniser/BaseTokeniser.d.ts +2 -1
  28. package/dist/tokeniser/BaseTokeniser.js +135 -2
  29. package/dist/tokeniser/CharTokeniser.d.ts +2 -2
  30. package/dist/tokeniser/CharTokeniser.js +10 -14
  31. package/dist/tokeniser/bpe.d.ts +2 -2
  32. package/dist/tokeniser/bpe.js +23 -32
  33. package/dist/tokeniser/type.d.ts +3 -2
  34. package/dist/training/DatasetBuilder.d.ts +17 -1
  35. package/dist/training/DatasetBuilder.js +2 -2
  36. package/dist/training/PreTrainer.js +1 -1
  37. package/dist/training/SFTTrainer.js +1 -1
  38. package/dist/training/tasks/ConversationTask.d.ts +9 -10
  39. package/dist/training/tasks/ConversationTask.js +15 -24
  40. package/dist/training/tasks/PretrainingTask.d.ts +4 -4
  41. package/dist/training/tasks/PretrainingTask.js +2 -2
  42. package/dist/training/tasks/StartSentenceTask.d.ts +4 -4
  43. package/dist/training/tasks/StartSentenceTask.js +3 -3
  44. package/dist/training/tasks/Task.d.ts +19 -12
  45. package/dist/training/tasks/Task.js +25 -30
  46. package/dist/training/tasks/TokenStore.d.ts +46 -0
  47. package/dist/training/tasks/TokenStore.js +215 -0
  48. package/dist/training/tasks/splitter.d.ts +2 -2
  49. package/dist/training/tasks/splitter.js +11 -11
  50. package/dist/training/validation.d.ts +4 -2
  51. package/dist/training/validation.js +21 -2
  52. package/dist/utilities/tokens.d.ts +2 -0
  53. package/dist/utilities/tokens.js +40 -0
  54. package/dist/utilities/topP.js +6 -0
  55. package/package.json +2 -1
package/dist/main.d.ts CHANGED
@@ -14,6 +14,8 @@ export { default as BPETokeniser } from './tokeniser/bpe';
14
14
  export { default as waitForModel } from './utilities/waitForModel';
15
15
  export { default as generateDatasetID } from './utilities/datasetID';
16
16
  export { default as loadTextData } from './data/textLoader';
17
+ export type { ConversationStream, ConversationCursor } from './data/stream';
18
+ export { MemoryConversationStream } from './data/stream';
17
19
  export type { DatasetMetadata, ModelMode } from './loader/types';
18
20
  export { default as Generator, type IGenerator } from './Generator';
19
21
  export { default as Evaluator } from './training/Evaluator';
@@ -25,6 +27,8 @@ export type { TrainingOptions, TrainingLogEntry } from './training/types';
25
27
  export type { GPTConfig } from './models/config';
26
28
  export { estimateParameterCount, estimateMemoryUsage, estimateTrainingMemoryUsage, estimateResources, validateConfig, } from './utilities/parameters';
27
29
  export { default as topP } from './utilities/topP';
30
+ export { sliceUint16Shards, sliceUint8Shards } from './utilities/tokens';
31
+ export { TokenStore, createTokenStore } from './training/tasks/TokenStore';
28
32
  export { Task, tokensFromTasks } from './training/tasks/Task';
29
33
  export declare const tasks: {
30
34
  PretrainingTask: typeof PretrainingTask;
package/dist/main.js CHANGED
@@ -1,16 +1,20 @@
1
- import { a as e, b as t, i as n, n as r, r as i, s as a, t as o, y as s } from "./main-DdsoTJAW.js";
2
- import c from "./tokeniser/CharTokeniser.js";
3
- import l from "./tokeniser/bpe.js";
4
- import { AdamWOptimizer as u } from "./training/AdamW.js";
5
- import d from "./utilities/topP.js";
6
- import f from "./training/Evaluator.js";
7
- import p from "./utilities/waitForModel.js";
8
- import m from "./utilities/datasetID.js";
9
- import h from "./data/textLoader.js";
10
- import { estimateMemoryUsage as g, estimateParameterCount as _, estimateResources as v, estimateTrainingMemoryUsage as y, validateConfig as b } from "./utilities/parameters.js";
11
- import { Task as x, tokensFromTasks as S } from "./training/tasks/Task.js";
12
- import { selectBackend as C } from "./backend.js";
13
- import w from "./utilities/performance.js";
14
- import T from "./checks/index.js";
15
- import { sentenceEmbeddings as E, sentenceEmbeddingsTensor as D } from "./utilities/sentences.js";
16
- export { u as AdamWOptimizer, l as BPETokeniser, c as CharTokeniser, f as Evaluator, a as Generator, t as Model, s as NanoGPT, x as Task, n as TeachableLLM, e as Trainer, T as checks, g as estimateMemoryUsage, _ as estimateParameterCount, v as estimateResources, y as estimateTrainingMemoryUsage, m as generateDatasetID, o as layers, h as loadTextData, r as ops, w as performanceTest, C as selectBackend, E as sentenceEmbeddings, D as sentenceEmbeddingsTensor, i as tasks, S as tokensFromTasks, d as topP, b as validateConfig, p as waitForModel };
1
+ import { _ as e, a as t, i as n, n as r, r as i, t as a, v as o } from "./main-BSaDGH7I.js";
2
+ import s from "./tokeniser/CharTokeniser.js";
3
+ import c from "./tokeniser/bpe.js";
4
+ import { AdamWOptimizer as l } from "./training/AdamW.js";
5
+ import u from "./utilities/topP.js";
6
+ import d from "./training/Evaluator.js";
7
+ import { TokenStore as f, createTokenStore as p } from "./training/tasks/TokenStore.js";
8
+ import { Task as m, tokensFromTasks as h } from "./training/tasks/Task.js";
9
+ import { t as g } from "./Trainer-DBsyWJ4s.js";
10
+ import _ from "./utilities/waitForModel.js";
11
+ import v from "./utilities/datasetID.js";
12
+ import { n as y } from "./stream-BjdpSNqB.js";
13
+ import b from "./data/textLoader.js";
14
+ import { estimateMemoryUsage as x, estimateParameterCount as S, estimateResources as C, estimateTrainingMemoryUsage as w, validateConfig as T } from "./utilities/parameters.js";
15
+ import { sliceUint16Shards as E, sliceUint8Shards as D } from "./utilities/tokens.js";
16
+ import { selectBackend as O } from "./backend.js";
17
+ import k from "./utilities/performance.js";
18
+ import A from "./checks/index.js";
19
+ import { sentenceEmbeddings as j, sentenceEmbeddingsTensor as M } from "./utilities/sentences.js";
20
+ export { l as AdamWOptimizer, c as BPETokeniser, s as CharTokeniser, d as Evaluator, t as Generator, y as MemoryConversationStream, o as Model, e as NanoGPT, m as Task, n as TeachableLLM, f as TokenStore, g as Trainer, A as checks, p as createTokenStore, x as estimateMemoryUsage, S as estimateParameterCount, C as estimateResources, w as estimateTrainingMemoryUsage, v as generateDatasetID, a as layers, b as loadTextData, r as ops, k as performanceTest, O as selectBackend, j as sentenceEmbeddings, M as sentenceEmbeddingsTensor, E as sliceUint16Shards, D as sliceUint8Shards, i as tasks, h as tokensFromTasks, u as topP, T as validateConfig, _ as waitForModel };
@@ -1,2 +1,2 @@
1
- import { y as e } from "../main-DdsoTJAW.js";
1
+ import { _ as e } from "../main-BSaDGH7I.js";
2
2
  export { e as default };
@@ -1,2 +1,2 @@
1
- import { v as e } from "../main-DdsoTJAW.js";
1
+ import { g as e } from "../main-BSaDGH7I.js";
2
2
  export { e as default };
@@ -1,2 +1,2 @@
1
- import { _ as e } from "../main-DdsoTJAW.js";
1
+ import { h as e } from "../main-BSaDGH7I.js";
2
2
  export { e as default };
@@ -1,2 +1,2 @@
1
- import { b as e } from "../main-DdsoTJAW.js";
1
+ import { v as e } from "../main-BSaDGH7I.js";
2
2
  export { e as default };