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