felixfoertsch/vorleser
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add more debug output for token count

Browse Source
This commit is contained in:
Felix Förtsch
2026-02-19 12:38:46 +01:00
parent 2774fb8379
commit def76b02b5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
VorleserMac/Services/KokoroPipeline.swift
View File

@@ -91,7 +91,7 @@ final class KokoroPipeline {
private func synthesizeTokenIds(_ tokenIdsAll: [Int], phonemeScalarCount: Int) throws -> [Float] { private func synthesizeTokenIds(_ tokenIdsAll: [Int], phonemeScalarCount: Int) throws -> [Float] {
let tokenIds = [0] + tokenIdsAll + [0] 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 inputIds = try MLMultiArray(shape: [1, 128], dataType: .int32)
let attention = try MLMultiArray(shape: [1, 128], dataType: .int32) let attention = try MLMultiArray(shape: [1, 128], dataType: .int32)