@genai-fi/nanogpt 1.2.3 → 1.2.4
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/inference/Beamer.js
CHANGED
|
@@ -101,10 +101,10 @@ var f = class {
|
|
|
101
101
|
});
|
|
102
102
|
}
|
|
103
103
|
}
|
|
104
|
-
let c = o.filter((e) => e.terminated).slice();
|
|
104
|
+
let c = o.filter((e) => e.terminated), f = c.slice();
|
|
105
105
|
a.sort((e, t) => t.score - e.score);
|
|
106
106
|
for (let e of a) {
|
|
107
|
-
if (
|
|
107
|
+
if (f.length >= r) break;
|
|
108
108
|
let t = e.terminatedByToken || e.terminatedByWhitespace, n = !e.dropLeadingWhitespaceTerminator, i = n ? e.parent.tokens.concat(e.token) : e.parent.tokens.slice(), a = e.terminatedByToken || e.dropLeadingWhitespaceTerminator ? e.parent.text : e.parent.text + e.tokenText, o = n ? this.appendTokenToContext(e.parent.context, e.parent.contextLength, e.token) : {
|
|
109
109
|
context: e.parent.context.clone(),
|
|
110
110
|
contextLength: e.parent.contextLength
|
|
@@ -121,7 +121,7 @@ var f = class {
|
|
|
121
121
|
attention: null,
|
|
122
122
|
loss: null,
|
|
123
123
|
multinomialRand: null
|
|
124
|
-
}),
|
|
124
|
+
}), f.push({
|
|
125
125
|
tokens: i,
|
|
126
126
|
score: e.score,
|
|
127
127
|
text: a,
|
|
@@ -132,8 +132,8 @@ var f = class {
|
|
|
132
132
|
candidates: u
|
|
133
133
|
});
|
|
134
134
|
}
|
|
135
|
-
if (u += a.length - c.length, o.forEach((e) => e.context.dispose()),
|
|
136
|
-
o =
|
|
135
|
+
if (u += a.length - f.length + c.length, o.forEach((e) => e.context.dispose()), f.length === 0) break;
|
|
136
|
+
o = f, n && Date.now() - l >= 40 && (n(o.slice(0, t.beams)), l = Date.now());
|
|
137
137
|
}
|
|
138
138
|
this.active = !1, u += o.length;
|
|
139
139
|
let f = o.slice().sort((e, t) => t.score - e.score), p = [], m = /* @__PURE__ */ new Set();
|
|
@@ -2,8 +2,8 @@ import { I as e } from "../dist-Da20xy8E.js";
|
|
|
2
2
|
//#region lib/inference/tokenisePrompt.ts
|
|
3
3
|
async function t(t, n, r, i) {
|
|
4
4
|
if (r) {
|
|
5
|
-
let a = r.length > 0 && r[r.length - 1].role === "text", o;
|
|
6
|
-
return
|
|
5
|
+
let a = r.length > 0 && (r[r.length - 1].role === "text" || r[r.length - 1].role === "assistant"), o;
|
|
6
|
+
return i?.nonConversational ? o = a && i?.continuation ? [t.bosToken, ...t.encode(r[r.length - 1].content)] : t.encodeAsSequence(r, !0) : a && i?.continuation ? (o = t.encodeConversation(r, !0), o[o.length - 1] === t.getSpecialTokenIndex("<|assistant_start|>") && o[o.length - 2] === t.getSpecialTokenIndex("<|assistant_end|>") && (o.pop(), o.pop())) : o = t.encodeConversation(r, !0), o.length > n && (o = o.slice(-n)), e([o], [1, o.length], "int32");
|
|
7
7
|
} else {
|
|
8
8
|
let n = i?.nonConversational ? void 0 : t.getSpecialTokenIndex("<|assistant_start|>"), r = n === void 0 ? [t.bosToken] : [t.bosToken, n];
|
|
9
9
|
return e([r], [1, r.length], "int32");
|