dnrops
/
transformers.js
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rename variable

This commit is contained in:
Joshua Lochner 2023-02-19 03:56:31 +02:00
parent 0b48767d1e
commit 6442932d53
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/models.js
View File

@ -409,9 +409,9 @@ class GPT2LMHeadModel extends GPT2PreTrainedModel {
});
beam.past_key_values = output.past_key_values;
let newTokenIds = sampler(output.logits);
let sampledTokens = sampler(output.logits);
for (let [newTokenId, logProb] of newTokenIds) {
for (let [newTokenId, logProb] of sampledTokens) {
// use previous beam as a starting point
let newBeam = { ...beam };