File tree Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Original file line number Diff line number Diff line change @@ -1797,6 +1797,18 @@ bool LanguageModelLookahead::computeScoresSparse(LanguageModelLookahead::Context
17971797 waitingLookaheadNodesByDepth_[nodes_[nodeIdx].depth ].push_back (std::make_pair (nodeIdx, (Score)0 ));
17981798 }
17991799
1800+ for (const Lm::WordScore& successor : successors) {
1801+ Score score = successor.score () * scale;
1802+
1803+ TokenNodeScore::const_iterator node = nodeForToken_.begin () + firstNodeForToken_[successor.token ()],
1804+ nodeEnd = nodeForToken_.begin () + firstNodeForToken_[successor.token () + 1 ];
1805+ for (; node != nodeEnd; ++node) {
1806+ Score endScore = score + (*node).second ;
1807+ LookaheadId nodeId = (*node).first ;
1808+ waitingLookaheadNodesByDepth_[nodes_[nodeId].depth ].push_back (std::make_pair (nodeId, endScore));
1809+ }
1810+ }
1811+
18001812 if (nodeRecombination_.empty ()) {
18011813 nodeRecombination_.resize (nEntries_, 0 );
18021814 }
You can’t perform that action at this time.
0 commit comments