@seanhogg/builderforce-memory-engine 2026.9.28 → 2026.9.30
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.
- package/dist/evermind/adapt.d.ts +13 -0
- package/dist/evermind/adapt.d.ts.map +1 -1
- package/dist/evermind/adapt.js +13 -1
- package/dist/evermind/adapt.js.map +1 -1
- package/dist/evermind/index.d.ts +3 -3
- package/dist/evermind/index.d.ts.map +1 -1
- package/dist/evermind/index.js +2 -2
- package/dist/evermind/index.js.map +1 -1
- package/dist/evermind/package.d.ts +16 -0
- package/dist/evermind/package.d.ts.map +1 -1
- package/dist/evermind/package.js +40 -14
- package/dist/evermind/package.js.map +1 -1
- package/dist/experience/approval.d.ts +16 -0
- package/dist/experience/approval.d.ts.map +1 -0
- package/dist/experience/approval.js +40 -0
- package/dist/experience/approval.js.map +1 -0
- package/dist/experience/corpus.d.ts +24 -0
- package/dist/experience/corpus.d.ts.map +1 -0
- package/dist/experience/corpus.js +90 -0
- package/dist/experience/corpus.js.map +1 -0
- package/dist/experience/index.d.ts +18 -0
- package/dist/experience/index.d.ts.map +1 -0
- package/dist/experience/index.js +8 -0
- package/dist/experience/index.js.map +1 -0
- package/dist/experience/overview.d.ts +52 -0
- package/dist/experience/overview.d.ts.map +1 -0
- package/dist/experience/overview.js +92 -0
- package/dist/experience/overview.js.map +1 -0
- package/dist/experience/schedule.d.ts +10 -0
- package/dist/experience/schedule.d.ts.map +1 -0
- package/dist/experience/schedule.js +22 -0
- package/dist/experience/schedule.js.map +1 -0
- package/dist/experience/store.d.ts +88 -0
- package/dist/experience/store.d.ts.map +1 -0
- package/dist/experience/store.js +128 -0
- package/dist/experience/store.js.map +1 -0
- package/dist/experience/trainOnce.d.ts +18 -0
- package/dist/experience/trainOnce.d.ts.map +1 -0
- package/dist/experience/trainOnce.js +74 -0
- package/dist/experience/trainOnce.js.map +1 -0
- package/dist/experience/types.d.ts +171 -0
- package/dist/experience/types.d.ts.map +1 -0
- package/dist/experience/types.js +22 -0
- package/dist/experience/types.js.map +1 -0
- package/dist/export/onnx.d.ts.map +1 -1
- package/dist/export/onnx.js +0 -8
- package/dist/export/onnx.js.map +1 -1
- package/dist/import/foreign/falcon_mamba.d.ts +15 -2
- package/dist/import/foreign/falcon_mamba.d.ts.map +1 -1
- package/dist/import/foreign/falcon_mamba.js +53 -36
- package/dist/import/foreign/falcon_mamba.js.map +1 -1
- package/dist/import/foreign/index.d.ts +5 -2
- package/dist/import/foreign/index.d.ts.map +1 -1
- package/dist/import/foreign/index.js +4 -2
- package/dist/import/foreign/index.js.map +1 -1
- package/dist/import/foreign/infer.d.ts +20 -0
- package/dist/import/foreign/infer.d.ts.map +1 -0
- package/dist/import/foreign/infer.js +58 -0
- package/dist/import/foreign/infer.js.map +1 -0
- package/dist/import/foreign/registry.d.ts.map +1 -1
- package/dist/import/foreign/registry.js +2 -1
- package/dist/import/foreign/registry.js.map +1 -1
- package/dist/import/foreign/vocab.d.ts +19 -0
- package/dist/import/foreign/vocab.d.ts.map +1 -0
- package/dist/import/foreign/vocab.js +40 -0
- package/dist/import/foreign/vocab.js.map +1 -0
- package/dist/import/index.d.ts +2 -2
- package/dist/import/index.d.ts.map +1 -1
- package/dist/import/index.js +1 -1
- package/dist/import/index.js.map +1 -1
- package/dist/import/safetensors.d.ts +1 -1
- package/dist/import/safetensors.js +11 -2
- package/dist/import/safetensors.js.map +1 -1
- package/dist/index.d.ts +5 -4
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +5 -3
- package/dist/index.js.map +1 -1
- package/dist/moe/moe_model.d.ts +2 -0
- package/dist/moe/moe_model.d.ts.map +1 -1
- package/dist/moe/moe_model.js +9 -7
- package/dist/moe/moe_model.js.map +1 -1
- package/dist/moe/moe_trainer.d.ts.map +1 -1
- package/dist/moe/moe_trainer.js +5 -8
- package/dist/moe/moe_trainer.js.map +1 -1
- package/dist/utils/rng.d.ts +7 -3
- package/dist/utils/rng.d.ts.map +1 -1
- package/dist/utils/rng.js +10 -6
- package/dist/utils/rng.js.map +1 -1
- package/package.json +1 -1
- package/src/evermind/adapt.ts +26 -1
- package/src/evermind/index.ts +3 -3
- package/src/evermind/package.ts +39 -12
- package/src/experience/approval.ts +49 -0
- package/src/experience/corpus.ts +103 -0
- package/src/experience/index.ts +35 -0
- package/src/experience/overview.ts +143 -0
- package/src/experience/schedule.ts +29 -0
- package/src/experience/store.ts +188 -0
- package/src/experience/trainOnce.ts +85 -0
- package/src/experience/types.ts +162 -0
- package/src/export/onnx.ts +0 -7
- package/src/import/foreign/falcon_mamba.ts +63 -40
- package/src/import/foreign/index.ts +5 -2
- package/src/import/foreign/infer.ts +62 -0
- package/src/import/foreign/registry.ts +2 -1
- package/src/import/foreign/vocab.ts +50 -0
- package/src/import/index.ts +5 -0
- package/src/import/safetensors.ts +9 -2
- package/src/index.ts +14 -1
- package/src/moe/moe_model.ts +9 -7
- package/src/moe/moe_trainer.ts +5 -8
- package/src/utils/rng.ts +10 -6
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/experience/types.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,iFAAiF;AACjF,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,EAAE,MAAM,CAAC;IACrB,sCAAsC;IACtC,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,gDAAgD;IAChD,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,sFAAsF;IACtF,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd;AAED,4FAA4F;AAC5F,MAAM,MAAM,MAAM,GACd;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,MAAM,EAAE,UAAU,CAAA;CAAE,GACrC;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,MAAM,EAAE,UAAU,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,OAAO,CAAA;CAAE,GACzE;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,CAAA;CAAE,CAAC;AAE/D,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,gDAAgD;IAChD,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,0EAA0E;IAC1E,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC5B;AAED,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,KAAK,EAAE,YAAY,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,4CAA4C;AAC5C,MAAM,WAAW,UAAU;IACzB,iCAAiC;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,mEAAmE;IACnE,KAAK,EAAE,MAAM,CAAC;IACd,gEAAgE;IAChE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,MAAM,MAAM,UAAU,GAAG;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAAG;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,CAAC;AAE7F,MAAM,MAAM,WAAW,GACnB;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,MAAM,EAAE,UAAU,CAAA;CAAE,GACrC;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,MAAM,EAAE,UAAU,CAAC;IAAC,KAAK,EAAE,UAAU,CAAA;CAAE,GAC3D;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,CAAA;CAAE,CAAC;AAE/D,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,WAAW,CAAC;IACpB,iDAAiD;IACjD,gBAAgB,EAAE,OAAO,CAAC;IAC1B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC5B;AAED,gEAAgE;AAChE,MAAM,MAAM,QAAQ,GAAG;IAAE,KAAK,EAAE,SAAS,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAAG;IAAE,KAAK,EAAE,KAAK,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,CAAC;AAE9G,0EAA0E;AAC1E,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE,QAAQ,CAAC;IACnB,gFAAgF;IAChF,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/B,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC3B;AAED,MAAM,WAAW,KAAK;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,MAAM,EAAE,UAAU,EAAE,CAAC;IACrB,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,YAAY,GAAG,IAAI,CAAC;CAC/B;AAED,MAAM,MAAM,SAAS,GAAG,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,SAAS,GAAG,QAAQ,CAAC;AAElF,MAAM,MAAM,WAAW,GAAG,IAAI,GAAG,UAAU,GAAG,QAAQ,GAAG,UAAU,GAAG,QAAQ,CAAC;AAE/E,gFAAgF;AAChF,MAAM,WAAW,UAAU;IACzB,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,WAAW,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,EAAE,EAAE,MAAM,CAAC;CACZ;AAED,MAAM,WAAW,GAAG;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,6BAA6B;IAC7B,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,SAAS,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,KAAK,EAAE,UAAU,EAAE,CAAC;CACrB;AAED;;;GAGG;AACH,MAAM,WAAW,UAAU;IACzB,sDAAsD;IACtD,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,wCAAwC;IACxC,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,qCAAqC;IACrC,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;CACZ;AAED,gEAAgE;AAChE,wBAAgB,eAAe,IAAI,MAAM,CAIxC"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* experience/types.ts — what Evermind has DONE, as opposed to what it knows.
|
|
3
|
+
*
|
|
4
|
+
* Facts live in `@seanhogg/builderforce-memory` (MemoryStore / EvermindCognition) and
|
|
5
|
+
* weights in the model; this is the third kind of learned item: experience.
|
|
6
|
+
* - an **Episode** is one demonstration — a person doing a task in a program, recorded
|
|
7
|
+
* as semantic steps (the UI element acted on and the value set, never raw keystrokes);
|
|
8
|
+
* - a **Skill** is what Train Once compiles from an episode — a procedure that runs
|
|
9
|
+
* again with new values, with approval gates on irreversible steps;
|
|
10
|
+
* - a **Run** is one execution of a skill, with its step-by-step audit trail.
|
|
11
|
+
*
|
|
12
|
+
* Plain data, JSON-shaped (camelCase, actions tagged by `kind`), so any recorder — the
|
|
13
|
+
* native Windows UI Automation one in Synapse included — produces it and any runtime
|
|
14
|
+
* persists it. Timestamps are epoch milliseconds.
|
|
15
|
+
*/
|
|
16
|
+
/** A short random id — unique enough for one person's store. */
|
|
17
|
+
export function newExperienceId() {
|
|
18
|
+
const bytes = new Uint8Array(8);
|
|
19
|
+
globalThis.crypto.getRandomValues(bytes);
|
|
20
|
+
return Array.from(bytes, (b) => b.toString(16).padStart(2, '0')).join('');
|
|
21
|
+
}
|
|
22
|
+
//# sourceMappingURL=types.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/experience/types.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AA8IH,gEAAgE;AAChE,MAAM,UAAU,eAAe;IAC7B,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IAChC,UAAU,CAAC,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IACzC,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAC5E,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"onnx.d.ts","sourceRoot":"","sources":["../../src/export/onnx.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;
|
|
1
|
+
{"version":3,"file":"onnx.d.ts","sourceRoot":"","sources":["../../src/export/onnx.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAiGvD,6CAA6C;AAC7C,wBAAgB,UAAU,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,GAAE;IAAE,YAAY,CAAC,EAAE,MAAM,CAAC;IAAC,eAAe,CAAC,EAAE,MAAM,CAAA;CAAO,GAAG,UAAU,CAsGrH"}
|
package/dist/export/onnx.js
CHANGED
|
@@ -62,14 +62,6 @@ class GraphBuilder {
|
|
|
62
62
|
return out;
|
|
63
63
|
}
|
|
64
64
|
}
|
|
65
|
-
/** Transpose a 2-D row-major matrix [rows, cols] → [cols, rows]. */
|
|
66
|
-
function transpose2d(data, rows, cols) {
|
|
67
|
-
const out = new Float32Array(data.length);
|
|
68
|
-
for (let r = 0; r < rows; r++)
|
|
69
|
-
for (let c = 0; c < cols; c++)
|
|
70
|
-
out[c * rows + r] = data[r * cols + c];
|
|
71
|
-
return out;
|
|
72
|
-
}
|
|
73
65
|
/** RMSNorm(x, gain) as graph ops. Returns the normalised value name. */
|
|
74
66
|
function rmsNorm(g, x, gainInit, eps, axes) {
|
|
75
67
|
const sq = g.op("Mul", [x, x], "rms_sq");
|
package/dist/export/onnx.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"onnx.js","sourceRoot":"","sources":["../../src/export/onnx.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AAGH,OAAO,EAAE,MAAM,EAAE,YAAY,EAAuC,MAAM,cAAc,CAAC;AACzF,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE5D,MAAM,KAAK,GAAG,EAAE,CAAC;AACjB,MAAM,UAAU,GAAG,CAAC,CAAC;AACrB,MAAM,UAAU,GAAG,CAAC,CAAC;AACrB,MAAM,UAAU,GAAG,CAAC,CAAC;AAErB,+BAA+B;AAC/B,MAAM,UAAU,GAAG,CAAC,CAAC;AACrB,MAAM,QAAQ,GAAG,CAAC,CAAC;AACnB,MAAM,WAAW,GAAG,CAAC,CAAC;AACtB,MAAM,SAAS,GAAG,CAAC,CAAC;AAwBpB,oFAAoF;AACpF,MAAM,YAAY;IACP,KAAK,GAAc,EAAE,CAAC;IACtB,KAAK,GAAc,EAAE,CAAC;IACvB,GAAG,GAAG,CAAC,CAAC;IAEhB,uCAAuC;IACvC,GAAG,CAAC,IAAY;QACd,OAAO,GAAG,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;IACjC,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,IAAc,EAAE,IAAkB;QACxD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,GAAG,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,IAAc,EAAE,MAAgB;QACtD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QACrE,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,CAAC,EAAU,EAAE,MAAgB,EAAE,OAAiB,EAAE,QAAgB,EAAE;QACtE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7E,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,gEAAgE;IAChE,EAAE,CAAC,EAAU,EAAE,MAAgB,EAAE,IAAY,EAAE,QAAgB,EAAE;QAC/D,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC3B,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC;QACpC,OAAO,GAAG,CAAC;IACb,CAAC;CACF;AAED,oEAAoE;AACpE,SAAS,WAAW,CAAC,IAAkB,EAAE,IAAY,EAAE,IAAY;IACjE,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC1C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE;QAAE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE;YAAE,GAAG,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC,CAAE,CAAC;IACtG,OAAO,GAAG,CAAC;AACb,CAAC;AAED,wEAAwE;AACxE,SAAS,OAAO,CAAC,CAAe,EAAE,CAAS,EAAE,QAAgB,EAAE,GAAW,EAAE,IAAY;IACtF,MAAM,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;IACzC,MAAM,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACnG,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,SAAS,CAAC,CAAC;IAC9C,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC;IACvC,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IAC3C,OAAO,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,GAAG,EAAE,QAAQ,CAAC,EAAE,OAAO,CAAC,CAAC;AAC/C,CAAC;AAED,4FAA4F;AAC5F,SAAS,GAAG,CACV,CAAe,EACf,CAAS,EACT,EAAU,EACV,EAAU,EACV,EAAU,EACV,EAAU;IAEV,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5F,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,SAAS,CAAC,EAAE,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;IAC9E,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC;IACvC,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5F,OAAO,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,SAAS,CAAC,EAAE,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;AACzE,CAAC;AAED,6CAA6C;AAC7C,MAAM,UAAU,UAAU,CAAC,EAAc,EAAE,OAA4D,EAAE;IACvG,MAAM,IAAI,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC;IACxB,MAAM,OAAO,GAAG,YAAY,CAAC,EAAE,CAAC,CAAC;IACjC,MAAM,MAAM,GAAG,IAAI,GAAG,CAAsB,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7E,MAAM,GAAG,GAAG,CAAC,IAAY,EAAe,EAAE;QACxC,MAAM,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC3B,IAAI,CAAC,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,IAAI,EAAE,CAAC,CAAC;QAC/D,OAAO,CAAC,CAAC;IACX,CAAC,CAAC;IAEF,MAAM,CAAC,GAAG,IAAI,YAAY,EAAE,CAAC;IAC7B,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC;IAE3F,mCAAmC;IACnC,MAAM,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,wBAAwB,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,CAAC;IAC9F,MAAM,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClE,MAAM,QAAQ,GAAG,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,0BAA0B;IAC/E,MAAM,KAAK,GAAG,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;IAEjD,+BAA+B;IAC/B,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,GAAG,EAAE,WAAW,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAEhG,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,UAAU,CAAC,EAAE,CAAC;QAExB,4EAA4E;QAC5E,MAAM,KAAK,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3F,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC;QACpD,qFAAqF;QACrF,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ;QAClD,MAAM,EAAE,GAAG,IAAI,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;YAAE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;gBAAE,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAE,CAAC;QAClG,MAAM,KAAK,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,mBAAmB,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAClE,MAAM,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU;QAClH,MAAM,IAAI,GAAG,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,KAAK,CAAC,EAAE,UAAU,EAAE;YACjD,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE;YACtC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;YAClD,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,EAAE;YACjD,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;YAC7C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;SAChD,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU;QACpH,MAAM,SAAS,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,YAAY,CAAC,CAAC;QAExD,4EAA4E;QAC5E,MAAM,IAAI,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,CAAC;QACxF,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC;QAE3D,MAAM,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,oBAAoB,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,CAAC;QAC7F,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ;QAC1G,MAAM,YAAY,GAAG,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,eAAe,CAAC,CAAC,CAAC,UAAU;QAChF,MAAM,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC9B,MAAM,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,YAAY,EAAE,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE;YACpD,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE;YACrC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE;YACxC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE;SACxC,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,CAAC,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU;QACpG,MAAM,KAAK,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,YAAY,EAAE,YAAY,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,UAAU;QAC5E,MAAM,OAAO,GAAG,CAAC,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU;QAE9H,gCAAgC;QAChC,MAAM,OAAO,GAAG,GAAG,CACjB,CAAC,EACD,OAAO,EACP,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,EACtF,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,EACnF,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,EACtF,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,CACvE,CAAC;QAEF,yDAAyD;QACzD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7E,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE;YACpC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE;YACrC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC,EAAE;SAC7C,CAAC,CAAC;QAEH,IAAI,MAAM,GAAG,OAAO,CAAC;QACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,MAAM,EAAE,GAAG,GAAG,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACnC,MAAM,EAAE,GAAG,GAAG,CACZ,CAAC,EACD,OAAO,EACP,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,EAClE,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,EAC/D,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,EAClE,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,CACnD,CAAC;YACF,MAAM,QAAQ,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,SAAS,CAAC,CAAC,CAAE,EAAE,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,4BAA4B;YAC3F,MAAM,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,SAAS,CAAC,CAAC;QACtD,CAAC;QAED,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC,EAAE,WAAW,CAAC,CAAC;IACpD,CAAC;IAED,gCAAgC;IAChC,MAAM,IAAI,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ;IACzG,CAAC,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;IAExC,OAAO,WAAW,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AACpC,CAAC;AAED,kFAAkF;AAElF,SAAS,SAAS,CAAC,CAAO;IACxB,MAAM,CAAC,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;IACpB,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;QACf,KAAK,GAAG;YACN,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;YACvB,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;YACrB,MAAM;QACR,KAAK,MAAM;YACT,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;YACxB,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK;gBAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,4BAA4B;YACrE,MAAM;QACR,KAAK,GAAG;YACN,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YACzB,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM;QACR,KAAK,GAAG;YACN,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;YAC1B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;YACrB,MAAM;IACV,CAAC;IACD,OAAO,CAAC,CAAC;AACX,CAAC;AAED,SAAS,SAAS,CAAC,CAAU;IAC3B,MAAM,CAAC,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM;QAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACzC,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO;QAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;IAClB,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK;QAAE,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IACpD,OAAO,CAAC,CAAC;AACX,CAAC;AAED,SAAS,WAAW,CAAC,CAAU;IAC7B,MAAM,CAAC,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI;QAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,sBAAsB;IAC9D,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC;IACxB,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;IACpB,IAAI,CAAC,CAAC,KAAK;QAAE,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK;YAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,wBAAwB;IAC9E,IAAI,CAAC,CAAC,GAAG;QAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,iCAAiC;IAC/D,OAAO,CAAC,CAAC;AACX,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC5B,MAAM,CAAC,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,sBAAsB;IACzC,OAAO,CAAC,CAAC;AACX,CAAC;AACD,SAAS,QAAQ,CAAC,CAAS;IACzB,MAAM,CAAC,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,sBAAsB;IACtC,OAAO,CAAC,CAAC;AACX,CAAC;AAED,iFAAiF;AACjF,SAAS,SAAS,CAAC,IAAY,EAAE,QAAgB,EAAE,IAAyB;IAC1E,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;IAChC,KAAK,MAAM,CAAC,IAAI,IAAI;QAAE,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1F,MAAM,UAAU,GAAG,IAAI,WAAW,EAAE,CAAC;IACrC,UAAU,CAAC,MAAM,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAC,YAAY;IAC5C,UAAU,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,QAAQ;IACtC,MAAM,IAAI,GAAG,IAAI,WAAW,EAAE,CAAC;IAC/B,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,cAAc;IAC3C,MAAM,EAAE,GAAG,IAAI,WAAW,EAAE,CAAC;IAC7B,EAAE,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACnB,EAAE,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACpB,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,WAAW,CAClB,CAAe,EACf,IAAkB,EAClB,IAAyD;IAEzD,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;IAChC,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK;QAAE,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IACxD,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAC5B,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK;QAAE,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1D,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,SAAS,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;IACxE,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,SAAS,CAAC,QAAQ,EAAE,UAAU,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAErF,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;IAChC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,iBAAiB;IACtC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAEvB,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;IAChC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAC5B,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,YAAY,IAAI,4BAA4B,CAAC,CAAC;IACnE,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,eAAe,IAAI,UAAU,CAAC,CAAC;IACpD,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACxB,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACxB,OAAO,KAAK,CAAC,MAAM,EAAE,CAAC;AACxB,CAAC"}
|
|
1
|
+
{"version":3,"file":"onnx.js","sourceRoot":"","sources":["../../src/export/onnx.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AAGH,OAAO,EAAE,MAAM,EAAE,YAAY,EAAuC,MAAM,cAAc,CAAC;AACzF,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE5D,MAAM,KAAK,GAAG,EAAE,CAAC;AACjB,MAAM,UAAU,GAAG,CAAC,CAAC;AACrB,MAAM,UAAU,GAAG,CAAC,CAAC;AACrB,MAAM,UAAU,GAAG,CAAC,CAAC;AAErB,+BAA+B;AAC/B,MAAM,UAAU,GAAG,CAAC,CAAC;AACrB,MAAM,QAAQ,GAAG,CAAC,CAAC;AACnB,MAAM,WAAW,GAAG,CAAC,CAAC;AACtB,MAAM,SAAS,GAAG,CAAC,CAAC;AAwBpB,oFAAoF;AACpF,MAAM,YAAY;IACP,KAAK,GAAc,EAAE,CAAC;IACtB,KAAK,GAAc,EAAE,CAAC;IACvB,GAAG,GAAG,CAAC,CAAC;IAEhB,uCAAuC;IACvC,GAAG,CAAC,IAAY;QACd,OAAO,GAAG,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;IACjC,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,IAAc,EAAE,IAAkB;QACxD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,GAAG,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,IAAc,EAAE,MAAgB;QACtD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QACrE,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,CAAC,EAAU,EAAE,MAAgB,EAAE,OAAiB,EAAE,QAAgB,EAAE;QACtE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7E,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,gEAAgE;IAChE,EAAE,CAAC,EAAU,EAAE,MAAgB,EAAE,IAAY,EAAE,QAAgB,EAAE;QAC/D,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC3B,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC;QACpC,OAAO,GAAG,CAAC;IACb,CAAC;CACF;AAED,wEAAwE;AACxE,SAAS,OAAO,CAAC,CAAe,EAAE,CAAS,EAAE,QAAgB,EAAE,GAAW,EAAE,IAAY;IACtF,MAAM,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;IACzC,MAAM,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACnG,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,SAAS,CAAC,CAAC;IAC9C,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC;IACvC,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IAC3C,OAAO,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,GAAG,EAAE,QAAQ,CAAC,EAAE,OAAO,CAAC,CAAC;AAC/C,CAAC;AAED,4FAA4F;AAC5F,SAAS,GAAG,CACV,CAAe,EACf,CAAS,EACT,EAAU,EACV,EAAU,EACV,EAAU,EACV,EAAU;IAEV,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5F,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,SAAS,CAAC,EAAE,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;IAC9E,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC;IACvC,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5F,OAAO,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,SAAS,CAAC,EAAE,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;AACzE,CAAC;AAED,6CAA6C;AAC7C,MAAM,UAAU,UAAU,CAAC,EAAc,EAAE,OAA4D,EAAE;IACvG,MAAM,IAAI,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC;IACxB,MAAM,OAAO,GAAG,YAAY,CAAC,EAAE,CAAC,CAAC;IACjC,MAAM,MAAM,GAAG,IAAI,GAAG,CAAsB,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7E,MAAM,GAAG,GAAG,CAAC,IAAY,EAAe,EAAE;QACxC,MAAM,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC3B,IAAI,CAAC,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,IAAI,EAAE,CAAC,CAAC;QAC/D,OAAO,CAAC,CAAC;IACX,CAAC,CAAC;IAEF,MAAM,CAAC,GAAG,IAAI,YAAY,EAAE,CAAC;IAC7B,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC;IAE3F,mCAAmC;IACnC,MAAM,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,wBAAwB,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,CAAC;IAC9F,MAAM,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClE,MAAM,QAAQ,GAAG,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,0BAA0B;IAC/E,MAAM,KAAK,GAAG,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;IAEjD,+BAA+B;IAC/B,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,GAAG,EAAE,WAAW,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAEhG,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,UAAU,CAAC,EAAE,CAAC;QAExB,4EAA4E;QAC5E,MAAM,KAAK,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3F,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC;QACpD,qFAAqF;QACrF,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ;QAClD,MAAM,EAAE,GAAG,IAAI,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;YAAE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;gBAAE,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAE,CAAC;QAClG,MAAM,KAAK,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,mBAAmB,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAClE,MAAM,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU;QAClH,MAAM,IAAI,GAAG,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,KAAK,CAAC,EAAE,UAAU,EAAE;YACjD,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE;YACtC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;YAClD,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,EAAE;YACjD,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;YAC7C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;SAChD,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU;QACpH,MAAM,SAAS,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,YAAY,CAAC,CAAC;QAExD,4EAA4E;QAC5E,MAAM,IAAI,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,CAAC;QACxF,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC;QAE3D,MAAM,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,oBAAoB,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,CAAC;QAC7F,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ;QAC1G,MAAM,YAAY,GAAG,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,eAAe,CAAC,CAAC,CAAC,UAAU;QAChF,MAAM,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC9B,MAAM,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,YAAY,EAAE,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE;YACpD,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE;YACrC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE;YACxC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE;SACxC,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,CAAC,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU;QACpG,MAAM,KAAK,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,YAAY,EAAE,YAAY,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,UAAU;QAC5E,MAAM,OAAO,GAAG,CAAC,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU;QAE9H,gCAAgC;QAChC,MAAM,OAAO,GAAG,GAAG,CACjB,CAAC,EACD,OAAO,EACP,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,EACtF,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,EACnF,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,EACtF,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,CACvE,CAAC;QAEF,yDAAyD;QACzD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7E,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE;YACpC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE;YACrC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC,EAAE;SAC7C,CAAC,CAAC;QAEH,IAAI,MAAM,GAAG,OAAO,CAAC;QACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,MAAM,EAAE,GAAG,GAAG,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACnC,MAAM,EAAE,GAAG,GAAG,CACZ,CAAC,EACD,OAAO,EACP,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,EAClE,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,EAC/D,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,EAClE,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,CACnD,CAAC;YACF,MAAM,QAAQ,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,SAAS,CAAC,CAAC,CAAE,EAAE,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,4BAA4B;YAC3F,MAAM,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,SAAS,CAAC,CAAC;QACtD,CAAC;QAED,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC,EAAE,WAAW,CAAC,CAAC;IACpD,CAAC;IAED,gCAAgC;IAChC,MAAM,IAAI,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ;IACzG,CAAC,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;IAExC,OAAO,WAAW,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AACpC,CAAC;AAED,kFAAkF;AAElF,SAAS,SAAS,CAAC,CAAO;IACxB,MAAM,CAAC,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;IACpB,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;QACf,KAAK,GAAG;YACN,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;YACvB,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;YACrB,MAAM;QACR,KAAK,MAAM;YACT,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;YACxB,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK;gBAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,4BAA4B;YACrE,MAAM;QACR,KAAK,GAAG;YACN,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YACzB,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM;QACR,KAAK,GAAG;YACN,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;YAC1B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;YACrB,MAAM;IACV,CAAC;IACD,OAAO,CAAC,CAAC;AACX,CAAC;AAED,SAAS,SAAS,CAAC,CAAU;IAC3B,MAAM,CAAC,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM;QAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACzC,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO;QAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;IAClB,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK;QAAE,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IACpD,OAAO,CAAC,CAAC;AACX,CAAC;AAED,SAAS,WAAW,CAAC,CAAU;IAC7B,MAAM,CAAC,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI;QAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,sBAAsB;IAC9D,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC;IACxB,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;IACpB,IAAI,CAAC,CAAC,KAAK;QAAE,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK;YAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,wBAAwB;IAC9E,IAAI,CAAC,CAAC,GAAG;QAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,iCAAiC;IAC/D,OAAO,CAAC,CAAC;AACX,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC5B,MAAM,CAAC,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,sBAAsB;IACzC,OAAO,CAAC,CAAC;AACX,CAAC;AACD,SAAS,QAAQ,CAAC,CAAS;IACzB,MAAM,CAAC,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,sBAAsB;IACtC,OAAO,CAAC,CAAC;AACX,CAAC;AAED,iFAAiF;AACjF,SAAS,SAAS,CAAC,IAAY,EAAE,QAAgB,EAAE,IAAyB;IAC1E,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;IAChC,KAAK,MAAM,CAAC,IAAI,IAAI;QAAE,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1F,MAAM,UAAU,GAAG,IAAI,WAAW,EAAE,CAAC;IACrC,UAAU,CAAC,MAAM,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAC,YAAY;IAC5C,UAAU,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,QAAQ;IACtC,MAAM,IAAI,GAAG,IAAI,WAAW,EAAE,CAAC;IAC/B,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,cAAc;IAC3C,MAAM,EAAE,GAAG,IAAI,WAAW,EAAE,CAAC;IAC7B,EAAE,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACnB,EAAE,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACpB,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,WAAW,CAClB,CAAe,EACf,IAAkB,EAClB,IAAyD;IAEzD,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;IAChC,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK;QAAE,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IACxD,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAC5B,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK;QAAE,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1D,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,SAAS,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;IACxE,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,SAAS,CAAC,QAAQ,EAAE,UAAU,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAErF,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;IAChC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,iBAAiB;IACtC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAEvB,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;IAChC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAC5B,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,YAAY,IAAI,4BAA4B,CAAC,CAAC;IACnE,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,eAAe,IAAI,UAAU,CAAC,CAAC;IACpD,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACxB,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACxB,OAAO,KAAK,CAAC,MAAM,EAAE,CAAC;AACxB,CAAC"}
|
|
@@ -1,6 +1,9 @@
|
|
|
1
1
|
/**
|
|
2
|
-
* import/foreign/falcon_mamba.ts — weight port for
|
|
3
|
-
* (`FalconMambaForCausalLM`, `model_type: "falcon_mamba"
|
|
2
|
+
* import/foreign/falcon_mamba.ts — weight port for the Mamba-1 / S6 family:
|
|
3
|
+
* Falcon-Mamba (`FalconMambaForCausalLM`, `model_type: "falcon_mamba"`) and the
|
|
4
|
+
* original Mamba (`MambaForCausalLM`, `model_type: "mamba"` — `state-spaces/mamba-*`).
|
|
5
|
+
* They share one tensor layout and one config vocabulary, so one rule table and one
|
|
6
|
+
* `describe` serve both ({@link mamba1Adapter}).
|
|
4
7
|
*
|
|
5
8
|
* Source layout confirmed against `tiiuae/falcon-mamba-7b`'s published
|
|
6
9
|
* `config.json`, `model.safetensors.index.json` (643 tensors = 10 per layer × 64
|
|
@@ -34,6 +37,8 @@
|
|
|
34
37
|
* • Linear weights are `[out, in]` row-major on both sides (no transpose).
|
|
35
38
|
*
|
|
36
39
|
* Fidelity caveats recorded on the plan rather than hidden:
|
|
40
|
+
* • (Falcon-Mamba only) the dt/B/C norms below. Plain Mamba has none, so its port
|
|
41
|
+
* is exact.
|
|
37
42
|
* • `use_bias: false` upstream, so `in_proj` / `x_proj` / `out_proj` ship no
|
|
38
43
|
* bias; this engine's blocks always own one, and they are filled with zeros
|
|
39
44
|
* (an exact no-op, not an approximation).
|
|
@@ -47,5 +52,13 @@
|
|
|
47
52
|
* tensor has no target and is reported as unmapped.
|
|
48
53
|
*/
|
|
49
54
|
import { type ForeignMambaAdapter } from "./adapter.js";
|
|
55
|
+
/** A Mamba-1 family adapter: the shared rule table under one checkpoint family's identity. */
|
|
56
|
+
export declare function mamba1Adapter(identity: {
|
|
57
|
+
id: string;
|
|
58
|
+
label: string;
|
|
59
|
+
modelTypes: readonly string[];
|
|
60
|
+
architectures: readonly string[];
|
|
61
|
+
}): ForeignMambaAdapter;
|
|
50
62
|
export declare const falconMambaAdapter: ForeignMambaAdapter;
|
|
63
|
+
export declare const mambaAdapter: ForeignMambaAdapter;
|
|
51
64
|
//# sourceMappingURL=falcon_mamba.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"falcon_mamba.d.ts","sourceRoot":"","sources":["../../../src/import/foreign/falcon_mamba.ts"],"names":[],"mappings":"AAAA
|
|
1
|
+
{"version":3,"file":"falcon_mamba.d.ts","sourceRoot":"","sources":["../../../src/import/foreign/falcon_mamba.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoDG;AAIH,OAAO,EAML,KAAK,mBAAmB,EAEzB,MAAM,cAAc,CAAC;AAwEtB,8FAA8F;AAC9F,wBAAgB,aAAa,CAAC,QAAQ,EAAE;IACtC,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,SAAS,MAAM,EAAE,CAAC;IAC9B,aAAa,EAAE,SAAS,MAAM,EAAE,CAAC;CAClC,GAAG,mBAAmB,CAuCtB;AAED,eAAO,MAAM,kBAAkB,qBAK7B,CAAC;AAEH,eAAO,MAAM,YAAY,qBAKvB,CAAC"}
|
|
@@ -1,6 +1,9 @@
|
|
|
1
1
|
/**
|
|
2
|
-
* import/foreign/falcon_mamba.ts — weight port for
|
|
3
|
-
* (`FalconMambaForCausalLM`, `model_type: "falcon_mamba"
|
|
2
|
+
* import/foreign/falcon_mamba.ts — weight port for the Mamba-1 / S6 family:
|
|
3
|
+
* Falcon-Mamba (`FalconMambaForCausalLM`, `model_type: "falcon_mamba"`) and the
|
|
4
|
+
* original Mamba (`MambaForCausalLM`, `model_type: "mamba"` — `state-spaces/mamba-*`).
|
|
5
|
+
* They share one tensor layout and one config vocabulary, so one rule table and one
|
|
6
|
+
* `describe` serve both ({@link mamba1Adapter}).
|
|
4
7
|
*
|
|
5
8
|
* Source layout confirmed against `tiiuae/falcon-mamba-7b`'s published
|
|
6
9
|
* `config.json`, `model.safetensors.index.json` (643 tensors = 10 per layer × 64
|
|
@@ -34,6 +37,8 @@
|
|
|
34
37
|
* • Linear weights are `[out, in]` row-major on both sides (no transpose).
|
|
35
38
|
*
|
|
36
39
|
* Fidelity caveats recorded on the plan rather than hidden:
|
|
40
|
+
* • (Falcon-Mamba only) the dt/B/C norms below. Plain Mamba has none, so its port
|
|
41
|
+
* is exact.
|
|
37
42
|
* • `use_bias: false` upstream, so `in_proj` / `x_proj` / `out_proj` ship no
|
|
38
43
|
* bias; this engine's blocks always own one, and they are filled with zeros
|
|
39
44
|
* (an exact no-op, not an approximation).
|
|
@@ -47,7 +52,6 @@
|
|
|
47
52
|
* tensor has no target and is reported as unmapped.
|
|
48
53
|
*/
|
|
49
54
|
import { expandFactor, readInnerSize, readTimeStepRank, requireInt, } from "./adapter.js";
|
|
50
|
-
const ID = "falcon_mamba";
|
|
51
55
|
/** `backbone.embeddings` (HF) vs `backbone.embedding` (Mamba-SSM native). */
|
|
52
56
|
const EMBEDDING_NAMES = ["backbone.embeddings.weight", "backbone.embedding.weight"];
|
|
53
57
|
function rules(a) {
|
|
@@ -92,40 +96,53 @@ function rules(a) {
|
|
|
92
96
|
out.push({ target: "final_norm", shape: [E], from: ["backbone.norm_f.weight"], op: { kind: "copy" } });
|
|
93
97
|
return out;
|
|
94
98
|
}
|
|
95
|
-
|
|
96
|
-
|
|
99
|
+
/** A Mamba-1 family adapter: the shared rule table under one checkpoint family's identity. */
|
|
100
|
+
export function mamba1Adapter(identity) {
|
|
101
|
+
const ID = identity.id;
|
|
102
|
+
return {
|
|
103
|
+
...identity,
|
|
104
|
+
describe(config) {
|
|
105
|
+
const dModel = requireInt(config, "hidden_size");
|
|
106
|
+
const numLayers = requireInt(config, "num_hidden_layers");
|
|
107
|
+
const vocabSize = requireInt(config, "vocab_size");
|
|
108
|
+
const dState = requireInt(config, "state_size");
|
|
109
|
+
const dConv = requireInt(config, "conv_kernel");
|
|
110
|
+
const dInner = readInnerSize(config, dModel);
|
|
111
|
+
const expand = expandFactor(dInner, dModel, identity.label);
|
|
112
|
+
const dtRank = readTimeStepRank(config, dModel);
|
|
113
|
+
const modelConfig = {
|
|
114
|
+
vocabSize,
|
|
115
|
+
dModel,
|
|
116
|
+
numLayers,
|
|
117
|
+
dState,
|
|
118
|
+
dConv,
|
|
119
|
+
expand,
|
|
120
|
+
// Every layer is a Mamba-1 mixer (the model's default schedule), but the
|
|
121
|
+
// dt rank and conv bias are per-block settings the shorthand can't carry.
|
|
122
|
+
defaultMamba1: { dtRank, biasConv: true },
|
|
123
|
+
};
|
|
124
|
+
const plan = {
|
|
125
|
+
adapter: ID,
|
|
126
|
+
rules: rules({ vocabSize, dModel, numLayers, dInner, dState, dConv, dtRank }),
|
|
127
|
+
// Nothing is partially consumed: every Mamba-1 tensor this plan
|
|
128
|
+
// touches is taken whole. `lm_head.weight` is untouched entirely and
|
|
129
|
+
// surfaces through `unmappedSources`.
|
|
130
|
+
discards: [],
|
|
131
|
+
};
|
|
132
|
+
return { modelConfig, plan };
|
|
133
|
+
},
|
|
134
|
+
};
|
|
135
|
+
}
|
|
136
|
+
export const falconMambaAdapter = mamba1Adapter({
|
|
137
|
+
id: "falcon_mamba",
|
|
97
138
|
label: "Falcon-Mamba (Mamba-1 / S6)",
|
|
98
139
|
modelTypes: ["falcon_mamba"],
|
|
99
140
|
architectures: ["FalconMambaForCausalLM"],
|
|
100
|
-
|
|
101
|
-
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
|
|
106
|
-
|
|
107
|
-
const expand = expandFactor(dInner, dModel, "Falcon-Mamba");
|
|
108
|
-
const dtRank = readTimeStepRank(config, dModel);
|
|
109
|
-
const modelConfig = {
|
|
110
|
-
vocabSize,
|
|
111
|
-
dModel,
|
|
112
|
-
numLayers,
|
|
113
|
-
dState,
|
|
114
|
-
dConv,
|
|
115
|
-
expand,
|
|
116
|
-
// Every layer is a Mamba-1 mixer (the model's default schedule), but the
|
|
117
|
-
// dt rank and conv bias are per-block settings the shorthand can't carry.
|
|
118
|
-
defaultMamba1: { dtRank, biasConv: true },
|
|
119
|
-
};
|
|
120
|
-
const plan = {
|
|
121
|
-
adapter: ID,
|
|
122
|
-
rules: rules({ vocabSize, dModel, numLayers, dInner, dState, dConv, dtRank }),
|
|
123
|
-
// Nothing is partially consumed: every Falcon-Mamba tensor this plan
|
|
124
|
-
// touches is taken whole. `lm_head.weight` is untouched entirely and
|
|
125
|
-
// surfaces through `unmappedSources`.
|
|
126
|
-
discards: [],
|
|
127
|
-
};
|
|
128
|
-
return { modelConfig, plan };
|
|
129
|
-
},
|
|
130
|
-
};
|
|
141
|
+
});
|
|
142
|
+
export const mambaAdapter = mamba1Adapter({
|
|
143
|
+
id: "mamba",
|
|
144
|
+
label: "Mamba (Mamba-1 / S6)",
|
|
145
|
+
modelTypes: ["mamba"],
|
|
146
|
+
architectures: ["MambaForCausalLM"],
|
|
147
|
+
});
|
|
131
148
|
//# sourceMappingURL=falcon_mamba.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"falcon_mamba.js","sourceRoot":"","sources":["../../../src/import/foreign/falcon_mamba.ts"],"names":[],"mappings":"AAAA
|
|
1
|
+
{"version":3,"file":"falcon_mamba.js","sourceRoot":"","sources":["../../../src/import/foreign/falcon_mamba.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoDG;AAIH,OAAO,EACL,YAAY,EACZ,aAAa,EACb,gBAAgB,EAChB,UAAU,GAIX,MAAM,cAAc,CAAC;AAEtB,6EAA6E;AAC7E,MAAM,eAAe,GAAG,CAAC,4BAA4B,EAAE,2BAA2B,CAAC,CAAC;AAEpF,SAAS,KAAK,CAAC,CAQd;IACC,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC;IAC/F,MAAM,GAAG,GAAe;QACtB,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;KACpF,CAAC;IAEF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,mBAAmB,CAAC,QAAQ,CAAC;QACvC,MAAM,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC;QACtB,GAAG,CAAC,IAAI,CACN,EAAE,MAAM,EAAE,GAAG,CAAC,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAClG;YACE,MAAM,EAAE,GAAG,CAAC,UAAU;YACtB,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;YACd,IAAI,EAAE,EAAE;YACR,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE;YAC9B,IAAI,EAAE,2EAA2E;SAClF,EACD;YACE,MAAM,EAAE,GAAG,CAAC,QAAQ;YACpB,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YACb,IAAI,EAAE,CAAC,GAAG,CAAC,gBAAgB,CAAC;YAC5B,EAAE,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE;YACxB,IAAI,EAAE,+EAA+E;SACtF,EACD,EAAE,MAAM,EAAE,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EACtF,EAAE,MAAM,EAAE,GAAG,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EACpG;YACE,MAAM,EAAE,GAAG,CAAC,SAAS;YACrB,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAClB,IAAI,EAAE,EAAE;YACR,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE;YAC9B,IAAI,EAAE,+CAA+C;SACtD,EACD,EAAE,MAAM,EAAE,GAAG,CAAC,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAC9F,EAAE,MAAM,EAAE,GAAG,CAAC,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EACzF,EAAE,MAAM,EAAE,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EACnF,EAAE,MAAM,EAAE,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAC5E,EAAE,MAAM,EAAE,GAAG,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAChG;YACE,MAAM,EAAE,GAAG,CAAC,WAAW;YACvB,KAAK,EAAE,CAAC,CAAC,CAAC;YACV,IAAI,EAAE,EAAE;YACR,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE;YAC9B,IAAI,EAAE,iDAAiD;SACxD,EACD;YACE,MAAM,EAAE,GAAG,CAAC,aAAa;YACzB,KAAK,EAAE,CAAC,CAAC,CAAC;YACV,IAAI,EAAE,CAAC,mBAAmB,CAAC,cAAc,CAAC;YAC1C,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;SACrB,CACF,CAAC;IACJ,CAAC;IAED,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,wBAAwB,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;IACvG,OAAO,GAAG,CAAC;AACb,CAAC;AAED,8FAA8F;AAC9F,MAAM,UAAU,aAAa,CAAC,QAK7B;IACC,MAAM,EAAE,GAAG,QAAQ,CAAC,EAAE,CAAC;IACvB,OAAO;QACL,GAAG,QAAQ;QAEX,QAAQ,CAAC,MAAqB;YAC5B,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;YACjD,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC;YAC1D,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YACnD,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAChD,MAAM,KAAK,GAAG,UAAU,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;YAChD,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAC7C,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,MAAM,GAAG,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAEhD,MAAM,WAAW,GAA2B;gBAC1C,SAAS;gBACT,MAAM;gBACN,SAAS;gBACT,MAAM;gBACN,KAAK;gBACL,MAAM;gBACN,yEAAyE;gBACzE,0EAA0E;gBAC1E,aAAa,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE;aAC1C,CAAC;YAEF,MAAM,IAAI,GAAa;gBACrB,OAAO,EAAE,EAAE;gBACX,KAAK,EAAE,KAAK,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;gBAC7E,gEAAgE;gBAChE,qEAAqE;gBACrE,sCAAsC;gBACtC,QAAQ,EAAE,EAAE;aACb,CAAC;YAEF,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QAC/B,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,aAAa,CAAC;IAC9C,EAAE,EAAE,cAAc;IAClB,KAAK,EAAE,6BAA6B;IACpC,UAAU,EAAE,CAAC,cAAc,CAAC;IAC5B,aAAa,EAAE,CAAC,wBAAwB,CAAC;CAC1C,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,YAAY,GAAG,aAAa,CAAC;IACxC,EAAE,EAAE,OAAO;IACX,KAAK,EAAE,sBAAsB;IAC7B,UAAU,EAAE,CAAC,OAAO,CAAC;IACrB,aAAa,EAAE,CAAC,kBAAkB,CAAC;CACpC,CAAC,CAAC"}
|
|
@@ -6,7 +6,7 @@
|
|
|
6
6
|
* to build plus a name-keyed set of weights, together with an explicit account of
|
|
7
7
|
* anything that did not map cleanly.
|
|
8
8
|
*
|
|
9
|
-
* Supported: Falcon-Mamba (Mamba-1) and Codestral-Mamba (Mamba-2). Transformer
|
|
9
|
+
* Supported: Mamba and Falcon-Mamba (Mamba-1) and Codestral-Mamba (Mamba-2). Transformer
|
|
10
10
|
* checkpoints are NOT portable at any tensor naming — distillation is the only
|
|
11
11
|
* route, and {@link foreignMambaAdapterFor} says so when it rejects one.
|
|
12
12
|
*/
|
|
@@ -16,6 +16,9 @@ export { applyPortedWeights } from "./apply.js";
|
|
|
16
16
|
export { executePortPlan, normaliseSourceName } from "./plan.js";
|
|
17
17
|
export type { PortDiscard, PortOp, PortPlan, PortRule, PortedTensors, SynthesisedTarget, } from "./plan.js";
|
|
18
18
|
export type { ForeignConfig, ForeignMambaAdapter, PortTarget } from "./adapter.js";
|
|
19
|
-
export { falconMambaAdapter } from "./falcon_mamba.js";
|
|
19
|
+
export { falconMambaAdapter, mambaAdapter, mamba1Adapter } from "./falcon_mamba.js";
|
|
20
|
+
export { mambaConfigFromTensors } from "./infer.js";
|
|
21
|
+
export { resizePortedVocab } from "./vocab.js";
|
|
22
|
+
export type { ResizeVocabOptions } from "./vocab.js";
|
|
20
23
|
export { codestralMambaAdapter } from "./codestral_mamba.js";
|
|
21
24
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/import/foreign/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EACL,sBAAsB,EACtB,sBAAsB,EACtB,gBAAgB,EAChB,2BAA2B,GAC5B,MAAM,eAAe,CAAC;AACvB,YAAY,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAEtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAEhD,OAAO,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,WAAW,CAAC;AACjE,YAAY,EACV,WAAW,EACX,MAAM,EACN,QAAQ,EACR,QAAQ,EACR,aAAa,EACb,iBAAiB,GAClB,MAAM,WAAW,CAAC;AAEnB,YAAY,EAAE,aAAa,EAAE,mBAAmB,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAEnF,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/import/foreign/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EACL,sBAAsB,EACtB,sBAAsB,EACtB,gBAAgB,EAChB,2BAA2B,GAC5B,MAAM,eAAe,CAAC;AACvB,YAAY,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAEtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAEhD,OAAO,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,WAAW,CAAC;AACjE,YAAY,EACV,WAAW,EACX,MAAM,EACN,QAAQ,EACR,QAAQ,EACR,aAAa,EACb,iBAAiB,GAClB,MAAM,WAAW,CAAC;AAEnB,YAAY,EAAE,aAAa,EAAE,mBAAmB,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAEnF,OAAO,EAAE,kBAAkB,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACpF,OAAO,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AACpD,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAC/C,YAAY,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AACrD,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC"}
|
|
@@ -6,13 +6,15 @@
|
|
|
6
6
|
* to build plus a name-keyed set of weights, together with an explicit account of
|
|
7
7
|
* anything that did not map cleanly.
|
|
8
8
|
*
|
|
9
|
-
* Supported: Falcon-Mamba (Mamba-1) and Codestral-Mamba (Mamba-2). Transformer
|
|
9
|
+
* Supported: Mamba and Falcon-Mamba (Mamba-1) and Codestral-Mamba (Mamba-2). Transformer
|
|
10
10
|
* checkpoints are NOT portable at any tensor naming — distillation is the only
|
|
11
11
|
* route, and {@link foreignMambaAdapterFor} says so when it rejects one.
|
|
12
12
|
*/
|
|
13
13
|
export { FOREIGN_MAMBA_ADAPTERS, foreignMambaAdapterFor, portForeignMamba, portForeignMambaSafetensors, } from "./registry.js";
|
|
14
14
|
export { applyPortedWeights } from "./apply.js";
|
|
15
15
|
export { executePortPlan, normaliseSourceName } from "./plan.js";
|
|
16
|
-
export { falconMambaAdapter } from "./falcon_mamba.js";
|
|
16
|
+
export { falconMambaAdapter, mambaAdapter, mamba1Adapter } from "./falcon_mamba.js";
|
|
17
|
+
export { mambaConfigFromTensors } from "./infer.js";
|
|
18
|
+
export { resizePortedVocab } from "./vocab.js";
|
|
17
19
|
export { codestralMambaAdapter } from "./codestral_mamba.js";
|
|
18
20
|
//# sourceMappingURL=index.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/import/foreign/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EACL,sBAAsB,EACtB,sBAAsB,EACtB,gBAAgB,EAChB,2BAA2B,GAC5B,MAAM,eAAe,CAAC;AAGvB,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAEhD,OAAO,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,WAAW,CAAC;AAYjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/import/foreign/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EACL,sBAAsB,EACtB,sBAAsB,EACtB,gBAAgB,EAChB,2BAA2B,GAC5B,MAAM,eAAe,CAAC;AAGvB,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAEhD,OAAO,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,WAAW,CAAC;AAYjE,OAAO,EAAE,kBAAkB,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACpF,OAAO,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AACpD,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAE/C,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* import/foreign/infer.ts — a Mamba-1 checkpoint's config, read off its tensors.
|
|
3
|
+
*
|
|
4
|
+
* `state-spaces/mamba-*` repositories publish `model.safetensors` with a
|
|
5
|
+
* `config.json` in the original Mamba-SSM vocabulary (`d_model`, `n_layer`, …)
|
|
6
|
+
* that no adapter reads, and a dropped safetensors file often arrives with no
|
|
7
|
+
* config at all. Every geometry fact the port needs is in the tensor shapes, so
|
|
8
|
+
* this derives the Hugging Face `MambaConfig` the {@link mambaAdapter} expects
|
|
9
|
+
* instead of anyone restating the derivation.
|
|
10
|
+
*/
|
|
11
|
+
import type { NamedTensor } from "../../export/tensors.js";
|
|
12
|
+
import type { ForeignConfig } from "./adapter.js";
|
|
13
|
+
/**
|
|
14
|
+
* The `MambaForCausalLM` config implied by a Mamba-1 checkpoint's tensors:
|
|
15
|
+
* embedding `[vocab, d_model]`, per-layer `in_proj [2·d_inner, d_model]`,
|
|
16
|
+
* `conv1d [d_inner, (1,) d_conv]`, `A_log [d_inner, d_state]` and
|
|
17
|
+
* `x_proj [dt_rank + 2·d_state, d_inner]`.
|
|
18
|
+
*/
|
|
19
|
+
export declare function mambaConfigFromTensors(tensors: readonly NamedTensor[]): ForeignConfig;
|
|
20
|
+
//# sourceMappingURL=infer.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"infer.d.ts","sourceRoot":"","sources":["../../../src/import/foreign/infer.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAOlD;;;;;GAKG;AACH,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,SAAS,WAAW,EAAE,GAAG,aAAa,CAoCrF"}
|
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* import/foreign/infer.ts — a Mamba-1 checkpoint's config, read off its tensors.
|
|
3
|
+
*
|
|
4
|
+
* `state-spaces/mamba-*` repositories publish `model.safetensors` with a
|
|
5
|
+
* `config.json` in the original Mamba-SSM vocabulary (`d_model`, `n_layer`, …)
|
|
6
|
+
* that no adapter reads, and a dropped safetensors file often arrives with no
|
|
7
|
+
* config at all. Every geometry fact the port needs is in the tensor shapes, so
|
|
8
|
+
* this derives the Hugging Face `MambaConfig` the {@link mambaAdapter} expects
|
|
9
|
+
* instead of anyone restating the derivation.
|
|
10
|
+
*/
|
|
11
|
+
import { normaliseSourceName } from "./plan.js";
|
|
12
|
+
function fail(detail) {
|
|
13
|
+
throw new Error(`import/foreign: cannot read a Mamba-1 config from these tensors — ${detail}`);
|
|
14
|
+
}
|
|
15
|
+
/**
|
|
16
|
+
* The `MambaForCausalLM` config implied by a Mamba-1 checkpoint's tensors:
|
|
17
|
+
* embedding `[vocab, d_model]`, per-layer `in_proj [2·d_inner, d_model]`,
|
|
18
|
+
* `conv1d [d_inner, (1,) d_conv]`, `A_log [d_inner, d_state]` and
|
|
19
|
+
* `x_proj [dt_rank + 2·d_state, d_inner]`.
|
|
20
|
+
*/
|
|
21
|
+
export function mambaConfigFromTensors(tensors) {
|
|
22
|
+
const byName = new Map(tensors.map((t) => [normaliseSourceName(t.name), t]));
|
|
23
|
+
const shape = (name) => byName.get(name)?.shape;
|
|
24
|
+
const emb = shape("backbone.embeddings.weight") ?? shape("backbone.embedding.weight");
|
|
25
|
+
if (!emb || emb.length !== 2)
|
|
26
|
+
fail("no backbone.embedding(s).weight [vocab, d_model]");
|
|
27
|
+
const [vocabSize, dModel] = emb;
|
|
28
|
+
let numLayers = 0;
|
|
29
|
+
while (byName.has(`backbone.layers.${numLayers}.mixer.in_proj.weight`))
|
|
30
|
+
numLayers++;
|
|
31
|
+
if (numLayers === 0)
|
|
32
|
+
fail("no backbone.layers.N.mixer.in_proj.weight");
|
|
33
|
+
const m = "backbone.layers.0.mixer";
|
|
34
|
+
const inProj = shape(`${m}.in_proj.weight`);
|
|
35
|
+
const conv = shape(`${m}.conv1d.weight`);
|
|
36
|
+
const aLog = shape(`${m}.A_log`);
|
|
37
|
+
const xProj = shape(`${m}.x_proj.weight`);
|
|
38
|
+
if (!conv || !aLog || !xProj)
|
|
39
|
+
fail(`layer 0 is missing conv1d.weight, A_log or x_proj.weight`);
|
|
40
|
+
const dInner = inProj[0] / 2;
|
|
41
|
+
const dState = aLog[1];
|
|
42
|
+
const dConv = conv[conv.length - 1];
|
|
43
|
+
const dtRank = xProj[0] - 2 * dState;
|
|
44
|
+
if (!Number.isInteger(dInner) || dtRank <= 0)
|
|
45
|
+
fail(`inconsistent shapes (in_proj ${inProj.join("×")}, x_proj ${xProj.join("×")})`);
|
|
46
|
+
return {
|
|
47
|
+
model_type: "mamba",
|
|
48
|
+
architectures: ["MambaForCausalLM"],
|
|
49
|
+
vocab_size: vocabSize,
|
|
50
|
+
hidden_size: dModel,
|
|
51
|
+
num_hidden_layers: numLayers,
|
|
52
|
+
intermediate_size: dInner,
|
|
53
|
+
state_size: dState,
|
|
54
|
+
conv_kernel: dConv,
|
|
55
|
+
time_step_rank: dtRank,
|
|
56
|
+
};
|
|
57
|
+
}
|
|
58
|
+
//# sourceMappingURL=infer.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"infer.js","sourceRoot":"","sources":["../../../src/import/foreign/infer.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAIH,OAAO,EAAE,mBAAmB,EAAE,MAAM,WAAW,CAAC;AAEhD,SAAS,IAAI,CAAC,MAAc;IAC1B,MAAM,IAAI,KAAK,CAAC,qEAAqE,MAAM,EAAE,CAAC,CAAC;AACjG,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,sBAAsB,CAAC,OAA+B;IACpE,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7E,MAAM,KAAK,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,KAAK,CAAC;IAExD,MAAM,GAAG,GAAG,KAAK,CAAC,4BAA4B,CAAC,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;IACtF,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;QAAE,IAAI,CAAC,kDAAkD,CAAC,CAAC;IACvF,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,GAAG,GAAuB,CAAC;IAEpD,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,OAAO,MAAM,CAAC,GAAG,CAAC,mBAAmB,SAAS,uBAAuB,CAAC;QAAE,SAAS,EAAE,CAAC;IACpF,IAAI,SAAS,KAAK,CAAC;QAAE,IAAI,CAAC,2CAA2C,CAAC,CAAC;IAEvE,MAAM,CAAC,GAAG,yBAAyB,CAAC;IACpC,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,iBAAiB,CAAE,CAAC;IAC7C,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IACzC,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACjC,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC1C,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK;QAAE,IAAI,CAAC,0DAA0D,CAAC,CAAC;IAE/F,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAE,GAAG,CAAC,CAAC;IAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC;IACxB,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAE,CAAC;IACrC,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,CAAE,GAAG,CAAC,GAAG,MAAM,CAAC;IACtC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,MAAM,IAAI,CAAC;QAAE,IAAI,CAAC,gCAAgC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAEnI,OAAO;QACL,UAAU,EAAE,OAAO;QACnB,aAAa,EAAE,CAAC,kBAAkB,CAAC;QACnC,UAAU,EAAE,SAAS;QACrB,WAAW,EAAE,MAAM;QACnB,iBAAiB,EAAE,SAAS;QAC5B,iBAAiB,EAAE,MAAM;QACzB,UAAU,EAAE,MAAM;QAClB,WAAW,EAAE,KAAK;QAClB,cAAc,EAAE,MAAM;KACvB,CAAC;AACJ,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../../src/import/foreign/registry.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AAEzE,OAAO,EAAmB,KAAK,WAAW,EAAE,KAAK,aAAa,EAAE,KAAK,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAG1G,OAAO,KAAK,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AAEvE,sEAAsE;AACtE,eAAO,MAAM,sBAAsB,EAAE,SAAS,mBAAmB,
|
|
1
|
+
{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../../src/import/foreign/registry.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AAEzE,OAAO,EAAmB,KAAK,WAAW,EAAE,KAAK,aAAa,EAAE,KAAK,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAG1G,OAAO,KAAK,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AAEvE,sEAAsE;AACtE,eAAO,MAAM,sBAAsB,EAAE,SAAS,mBAAmB,EAIhE,CAAC;AAEF,kFAAkF;AAClF,MAAM,WAAW,gBAAiB,SAAQ,aAAa;IACrD,yFAAyF;IACzF,WAAW,EAAE,sBAAsB,CAAC;CACrC;AAED,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,CAAC;AAkC/C;;;;GAIG;AACH,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,aAAa,GAAG,mBAAmB,CAoCjF;AAID;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,gBAAgB,CAIhG;AAED,+DAA+D;AAC/D,wBAAgB,2BAA2B,CACzC,MAAM,EAAE,aAAa,EACrB,KAAK,EAAE,UAAU,GAChB,gBAAgB,CAElB"}
|
|
@@ -9,10 +9,11 @@
|
|
|
9
9
|
*/
|
|
10
10
|
import { safetensorsToTensors } from "../safetensors.js";
|
|
11
11
|
import { executePortPlan } from "./plan.js";
|
|
12
|
-
import { falconMambaAdapter } from "./falcon_mamba.js";
|
|
12
|
+
import { falconMambaAdapter, mambaAdapter } from "./falcon_mamba.js";
|
|
13
13
|
import { codestralMambaAdapter } from "./codestral_mamba.js";
|
|
14
14
|
/** Every foreign SSM architecture this engine can warm-start from. */
|
|
15
15
|
export const FOREIGN_MAMBA_ADAPTERS = [
|
|
16
|
+
mambaAdapter,
|
|
16
17
|
falconMambaAdapter,
|
|
17
18
|
codestralMambaAdapter,
|
|
18
19
|
];
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../../src/import/foreign/registry.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAIH,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAgE,MAAM,WAAW,CAAC;AAC1G,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;
|
|
1
|
+
{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../../src/import/foreign/registry.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAIH,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAgE,MAAM,WAAW,CAAC;AAC1G,OAAO,EAAE,kBAAkB,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACrE,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAG7D,sEAAsE;AACtE,MAAM,CAAC,MAAM,sBAAsB,GAAmC;IACpE,YAAY;IACZ,kBAAkB;IAClB,qBAAqB;CACtB,CAAC;AAUF,gFAAgF;AAEhF,SAAS,eAAe,CAAC,MAAqB;IAC5C,MAAM,GAAG,GAAG,MAAM,CAAC,eAAe,CAAC,CAAC;IACpC,OAAO,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AACzF,CAAC;AAED,SAAS,WAAW,CAAC,MAAqB;IACxC,MAAM,GAAG,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;IACjC,OAAO,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC;AACnD,CAAC;AAED;;;;;GAKG;AACH,SAAS,oBAAoB,CAAC,aAAuB,EAAE,SAA6B;IAClF,MAAM,QAAQ,GAAG,CAAC,GAAG,aAAa,EAAE,SAAS,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;IAC7E,IAAI,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IACzE,OAAO,kGAAkG,CAAC,IAAI,CAC5G,QAAQ,CACT,CAAC;AACJ,CAAC;AAED,SAAS,aAAa;IACpB,OAAO,sBAAsB,CAAC,GAAG,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,oBAAoB,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAC3G,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,sBAAsB,CAAC,MAAqB;IAC1D,MAAM,aAAa,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IAC9C,MAAM,SAAS,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;IAEtC,MAAM,MAAM,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAC/C,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAC7D,CAAC;IACF,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IAE1B,MAAM,MAAM,GAAG,SAAS;QACtB,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QACtE,CAAC,CAAC,SAAS,CAAC;IACd,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IAE1B,MAAM,IAAI,GACR,aAAa,CAAC,MAAM,GAAG,CAAC;QACtB,CAAC,CAAC,kBAAkB,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG;QAC/C,CAAC,CAAC,SAAS;YACT,CAAC,CAAC,eAAe,SAAS,GAAG;YAC7B,CAAC,CAAC,6DAA6D,CAAC;IAEtE,IAAI,oBAAoB,CAAC,aAAa,EAAE,SAAS,CAAC,EAAE,CAAC;QACnD,MAAM,IAAI,KAAK,CACb,mBAAmB,IAAI,uDAAuD;YAC5E,oFAAoF;YACpF,qFAAqF;YACrF,4EAA4E;YAC5E,4EAA4E;YAC5E,yCAAyC,aAAa,EAAE,GAAG,CAC9D,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,KAAK,CACb,4CAA4C,IAAI,gBAAgB,aAAa,EAAE,IAAI;QACjF,0FAA0F,CAC7F,CAAC;AACJ,CAAC;AAED,gFAAgF;AAEhF;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAAC,MAAqB,EAAE,OAAsB;IAC5E,MAAM,OAAO,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;IAC/C,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IACvD,OAAO,EAAE,GAAG,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,WAAW,EAAE,CAAC;AAC5D,CAAC;AAED,+DAA+D;AAC/D,MAAM,UAAU,2BAA2B,CACzC,MAAqB,EACrB,KAAiB;IAEjB,OAAO,gBAAgB,CAAC,MAAM,EAAE,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC;AAC/D,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* import/foreign/vocab.ts — fit a ported checkpoint to a different vocabulary size.
|
|
3
|
+
*
|
|
4
|
+
* A ported model keeps its source tokenizer's vocabulary. To pair it with another
|
|
5
|
+
* tokenizer the embedding table (tied to the LM head) must have that tokenizer's row
|
|
6
|
+
* count: surplus rows are dropped, missing rows are drawn small and random so the
|
|
7
|
+
* new tokens start near the embedding cloud's centre. The new rows are recorded on
|
|
8
|
+
* the port as synthesised, never passed off as ported.
|
|
9
|
+
*/
|
|
10
|
+
import type { PortedCheckpoint } from "./registry.js";
|
|
11
|
+
export interface ResizeVocabOptions {
|
|
12
|
+
/** Standard deviation of new rows. Default 0.02, the usual embedding init. */
|
|
13
|
+
std?: number;
|
|
14
|
+
/** Seed for the new rows, so the same resize produces the same checkpoint. Default 1. */
|
|
15
|
+
seed?: number;
|
|
16
|
+
}
|
|
17
|
+
/** `ported` with an embedding of `vocabSize` rows and a model config to match. */
|
|
18
|
+
export declare function resizePortedVocab(ported: PortedCheckpoint, vocabSize: number, opts?: ResizeVocabOptions): PortedCheckpoint;
|
|
19
|
+
//# sourceMappingURL=vocab.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"vocab.d.ts","sourceRoot":"","sources":["../../../src/import/foreign/vocab.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAGH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAEtD,MAAM,WAAW,kBAAkB;IACjC,8EAA8E;IAC9E,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,yFAAyF;IACzF,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,kFAAkF;AAClF,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,gBAAgB,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,GAAE,kBAAuB,GAAG,gBAAgB,CA4B9H"}
|
|
@@ -0,0 +1,40 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* import/foreign/vocab.ts — fit a ported checkpoint to a different vocabulary size.
|
|
3
|
+
*
|
|
4
|
+
* A ported model keeps its source tokenizer's vocabulary. To pair it with another
|
|
5
|
+
* tokenizer the embedding table (tied to the LM head) must have that tokenizer's row
|
|
6
|
+
* count: surplus rows are dropped, missing rows are drawn small and random so the
|
|
7
|
+
* new tokens start near the embedding cloud's centre. The new rows are recorded on
|
|
8
|
+
* the port as synthesised, never passed off as ported.
|
|
9
|
+
*/
|
|
10
|
+
import { SeededRng, gaussianArray } from "../../utils/rng.js";
|
|
11
|
+
/** `ported` with an embedding of `vocabSize` rows and a model config to match. */
|
|
12
|
+
export function resizePortedVocab(ported, vocabSize, opts = {}) {
|
|
13
|
+
if (!Number.isInteger(vocabSize) || vocabSize <= 0) {
|
|
14
|
+
throw new Error(`import/foreign: vocabulary size must be a positive integer, got ${vocabSize}`);
|
|
15
|
+
}
|
|
16
|
+
const from = ported.modelConfig.vocabSize;
|
|
17
|
+
if (vocabSize === from)
|
|
18
|
+
return ported;
|
|
19
|
+
const emb = ported.weights.get("embedding");
|
|
20
|
+
if (!emb)
|
|
21
|
+
throw new Error('import/foreign: the port has no "embedding" to resize');
|
|
22
|
+
const dModel = ported.modelConfig.dModel;
|
|
23
|
+
const resized = new Float32Array(vocabSize * dModel);
|
|
24
|
+
resized.set(emb.subarray(0, Math.min(from, vocabSize) * dModel));
|
|
25
|
+
const synthesisedTargets = [...ported.synthesisedTargets];
|
|
26
|
+
if (vocabSize > from) {
|
|
27
|
+
const std = opts.std ?? 0.02;
|
|
28
|
+
const rng = new SeededRng(opts.seed ?? 1);
|
|
29
|
+
resized.set(gaussianArray((vocabSize - from) * dModel, std, () => rng.next()), from * dModel);
|
|
30
|
+
synthesisedTargets.push({
|
|
31
|
+
target: "embedding",
|
|
32
|
+
value: std,
|
|
33
|
+
reason: `rows ${from}..${vocabSize - 1} drawn from N(0, ${std}²) to reach a ${vocabSize}-token vocabulary`,
|
|
34
|
+
});
|
|
35
|
+
}
|
|
36
|
+
const weights = new Map(ported.weights);
|
|
37
|
+
weights.set("embedding", resized);
|
|
38
|
+
return { ...ported, weights, synthesisedTargets, modelConfig: { ...ported.modelConfig, vocabSize } };
|
|
39
|
+
}
|
|
40
|
+
//# sourceMappingURL=vocab.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"vocab.js","sourceRoot":"","sources":["../../../src/import/foreign/vocab.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAU9D,kFAAkF;AAClF,MAAM,UAAU,iBAAiB,CAAC,MAAwB,EAAE,SAAiB,EAAE,OAA2B,EAAE;IAC1G,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;QACnD,MAAM,IAAI,KAAK,CAAC,mEAAmE,SAAS,EAAE,CAAC,CAAC;IAClG,CAAC;IACD,MAAM,IAAI,GAAG,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC;IAC1C,IAAI,SAAS,KAAK,IAAI;QAAE,OAAO,MAAM,CAAC;IAEtC,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC5C,IAAI,CAAC,GAAG;QAAE,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;IACnF,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC;IAEzC,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,SAAS,GAAG,MAAM,CAAC,CAAC;IACrD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC;IACjE,MAAM,kBAAkB,GAAG,CAAC,GAAG,MAAM,CAAC,kBAAkB,CAAC,CAAC;IAC1D,IAAI,SAAS,GAAG,IAAI,EAAE,CAAC;QACrB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC;QAC7B,MAAM,GAAG,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC1C,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,GAAG,MAAM,CAAC,CAAC;QAC9F,kBAAkB,CAAC,IAAI,CAAC;YACtB,MAAM,EAAE,WAAW;YACnB,KAAK,EAAE,GAAG;YACV,MAAM,EAAE,QAAQ,IAAI,KAAK,SAAS,GAAG,CAAC,oBAAoB,GAAG,iBAAiB,SAAS,mBAAmB;SAC3G,CAAC,CAAC;IACL,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACxC,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;IAClC,OAAO,EAAE,GAAG,MAAM,EAAE,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,EAAE,GAAG,MAAM,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE,CAAC;AACvG,CAAC"}
|
package/dist/import/index.d.ts
CHANGED
|
@@ -13,6 +13,6 @@
|
|
|
13
13
|
export { safetensorsToTensors } from "./safetensors.js";
|
|
14
14
|
export { importEvermind, importEvermindTensors, inferArchFromTensors } from "./evermind.js";
|
|
15
15
|
export type { ImportOptions } from "./evermind.js";
|
|
16
|
-
export { FOREIGN_MAMBA_ADAPTERS, foreignMambaAdapterFor, portForeignMamba, portForeignMambaSafetensors, applyPortedWeights, executePortPlan, normaliseSourceName, falconMambaAdapter, codestralMambaAdapter, } from "./foreign/index.js";
|
|
17
|
-
export type { ForeignConfig, ForeignMambaAdapter, PortTarget, PortDiscard, PortOp, PortPlan, PortRule, PortedCheckpoint, PortedTensors, SynthesisedTarget, } from "./foreign/index.js";
|
|
16
|
+
export { FOREIGN_MAMBA_ADAPTERS, foreignMambaAdapterFor, portForeignMamba, portForeignMambaSafetensors, applyPortedWeights, executePortPlan, normaliseSourceName, falconMambaAdapter, mambaAdapter, mamba1Adapter, codestralMambaAdapter, mambaConfigFromTensors, resizePortedVocab, } from "./foreign/index.js";
|
|
17
|
+
export type { ForeignConfig, ForeignMambaAdapter, PortTarget, PortDiscard, PortOp, PortPlan, PortRule, PortedCheckpoint, PortedTensors, SynthesisedTarget, ResizeVocabOptions, } from "./foreign/index.js";
|
|
18
18
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/import/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAC5F,YAAY,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAGnD,OAAO,EACL,sBAAsB,EACtB,sBAAsB,EACtB,gBAAgB,EAChB,2BAA2B,EAC3B,kBAAkB,EAClB,eAAe,EACf,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/import/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAC5F,YAAY,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAGnD,OAAO,EACL,sBAAsB,EACtB,sBAAsB,EACtB,gBAAgB,EAChB,2BAA2B,EAC3B,kBAAkB,EAClB,eAAe,EACf,mBAAmB,EACnB,kBAAkB,EAClB,YAAY,EACZ,aAAa,EACb,qBAAqB,EACrB,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,oBAAoB,CAAC;AAC5B,YAAY,EACV,aAAa,EACb,mBAAmB,EACnB,UAAU,EACV,WAAW,EACX,MAAM,EACN,QAAQ,EACR,QAAQ,EACR,gBAAgB,EAChB,aAAa,EACb,iBAAiB,EACjB,kBAAkB,GACnB,MAAM,oBAAoB,CAAC"}
|
package/dist/import/index.js
CHANGED
|
@@ -13,5 +13,5 @@
|
|
|
13
13
|
export { safetensorsToTensors } from "./safetensors.js";
|
|
14
14
|
export { importEvermind, importEvermindTensors, inferArchFromTensors } from "./evermind.js";
|
|
15
15
|
// ── Foreign checkpoints (Falcon-Mamba / Codestral-Mamba → HybridMambaModel) ────
|
|
16
|
-
export { FOREIGN_MAMBA_ADAPTERS, foreignMambaAdapterFor, portForeignMamba, portForeignMambaSafetensors, applyPortedWeights, executePortPlan, normaliseSourceName, falconMambaAdapter, codestralMambaAdapter, } from "./foreign/index.js";
|
|
16
|
+
export { FOREIGN_MAMBA_ADAPTERS, foreignMambaAdapterFor, portForeignMamba, portForeignMambaSafetensors, applyPortedWeights, executePortPlan, normaliseSourceName, falconMambaAdapter, mambaAdapter, mamba1Adapter, codestralMambaAdapter, mambaConfigFromTensors, resizePortedVocab, } from "./foreign/index.js";
|
|
17
17
|
//# sourceMappingURL=index.js.map
|
package/dist/import/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/import/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAG5F,kFAAkF;AAClF,OAAO,EACL,sBAAsB,EACtB,sBAAsB,EACtB,gBAAgB,EAChB,2BAA2B,EAC3B,kBAAkB,EAClB,eAAe,EACf,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/import/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAG5F,kFAAkF;AAClF,OAAO,EACL,sBAAsB,EACtB,sBAAsB,EACtB,gBAAgB,EAChB,2BAA2B,EAC3B,kBAAkB,EAClB,eAAe,EACf,mBAAmB,EACnB,kBAAkB,EAClB,YAAY,EACZ,aAAa,EACb,qBAAqB,EACrB,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,oBAAoB,CAAC"}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* The exact inverse of {@link ../export/safetensors.ts}: parse the 8-byte
|
|
5
5
|
* little-endian u64 header length, the JSON header (name → {dtype, shape,
|
|
6
6
|
* data_offsets}), then decode each tensor's raw bytes into a Float32Array.
|
|
7
|
-
* F32 and
|
|
7
|
+
* F32, F16 and BF16 are supported (both halves are widened to f32 on read). This is the reader
|
|
8
8
|
* half of the warm-start / weight-port path — export was one-directional before.
|
|
9
9
|
*/
|
|
10
10
|
import type { NamedTensor } from "../export/tensors.js";
|