mii/AzooKeyKanaKanjiConverter
1
0
Fork 0
mirror of https://github.com/mii443/AzooKeyKanaKanjiConverter.git synced 2025-08-22 15:05:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: swiftlint

Browse Source
This commit is contained in:
Miwa / Ensan
2024-08-07 23:10:51 +09:00
parent 068585c2a5
commit 9d31eeee66
5 changed files with 11 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
Sources/KanaKanjiConverterModule/Zenz/ZenzContext.swift
View File

@ -47,7 +47,6 @@ struct FixedSizeHeap<Element: Comparable> {
}
}
enum ZenzError: LocalizedError {
case couldNotLoadModel(path: String)
case couldNotLoadContext
@ -241,7 +240,7 @@ class ZenzContext {
// Heapからソートして結果を取り出す
return minHeap.unordered.sorted { $0.value > $1.value }.map { ($0.character, $0.value / exp_sum) }
}
func evaluate_candidate(input: String, candidate: Candidate, versionDependentConfig: ConvertRequestOptions.ZenzaiVersionDependentMode) -> CandidateEvaluationResult {
print("Evaluate", candidate)
// For zenz-v1 model, \u{EE00} is a token used for 'start query', and \u{EE01} is a token used for 'start answer'
@ -288,7 +287,7 @@ class ZenzContext {
static func < (lhs: AlternativeHighProbToken, rhs: AlternativeHighProbToken) -> Bool {
lhs.probabilityRatioToMaxProb < rhs.probabilityRatioToMaxProb
}
var token: llama_token
var constraint: [UInt8]
// 最大probabilityに対しての割合
@ -304,7 +303,7 @@ class ZenzContext {
static func < (lhs: TokenAndExpLogit, rhs: TokenAndExpLogit) -> Bool {
lhs.expLogit < rhs.expLogit
}
var token: llama_token
var expLogit: Float
}
@ -390,7 +389,7 @@ class ZenzContext {
var swiftTokens: [llama_token] = if tokenCount < 0 {
[llama_token_bos(model)]
} else {
(0..<tokenCount).map{tokens[Int($0)]}
(0..<tokenCount).map {tokens[Int($0)]}
}
tokens.deallocate()
if add_eos {