Skip to content

Commit 28a931c

Browse files
author
Aegis-AI
committed
feat: replace noisy debug print with structured acceptance metrics in MTPTokenIterator
1 parent 0462ec4 commit 28a931c

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

Libraries/MLXLMCommon/Evaluate.swift

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1094,6 +1094,8 @@ public struct MTPTokenIterator: TokenIteratorProtocol {
10941094
private var pendingIndex = 0
10951095

10961096
// Internal metrics
1097+
public var acceptedDraftTokens: Int = 0
1098+
public var totalDraftTokens: Int = 0
10971099
var promptPrefillTime: TimeInterval = 0.0
10981100

10991101
/// Initialize a `MTPTokenIterator` with the given input.
@@ -1317,10 +1319,8 @@ public struct MTPTokenIterator: TokenIteratorProtocol {
13171319
processor?.didSample(token: finalTokenOut)
13181320
pendingTokens.append(finalTokenOut.item(Int.self))
13191321
}
1320-
1321-
if draftTokens.count > 0 {
1322-
print(" [MTP Debug] Drafts: \(draftTokensList), Target: \(mainTokensList), Accepted: \(accepted)")
1323-
}
1322+
self.acceptedDraftTokens += accepted
1323+
self.totalDraftTokens += draftTokens.count
13241324

13251325
// Rewind caches for rejected tokens
13261326
let rejectedCount = draftTokens.count - accepted

0 commit comments

Comments
 (0)