add more debug output for token count

2026-02-19 12:38:46 +01:00
parent 2774fb8379
commit def76b02b5
1 changed files with 1 additions and 1 deletions
--- a/VorleserMac/Services/KokoroPipeline.swift
+++ b/VorleserMac/Services/KokoroPipeline.swift
@@ -91,7 +91,7 @@ final class KokoroPipeline {

    private func synthesizeTokenIds(_ tokenIdsAll: [Int], phonemeScalarCount: Int) throws -> [Float] {
        let tokenIds = [0] + tokenIdsAll + [0]
-        print("Token count: \(tokenIds.count)")
+        print("Token count: \(tokenIds.count) (input: \(tokenIdsAll.count) + 2 for start/end)")

        let inputIds = try MLMultiArray(shape: [1, 128], dataType: .int32)
        let attention = try MLMultiArray(shape: [1, 128], dataType: .int32)