jake/vOOice
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Move LocalModel to TranscriptionModel file

Browse Source
This commit is contained in:
Beingpax 2025-07-23 09:30:44 +05:45
parent 0868228617
commit adb51193c6
2 changed files with 42 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
VoiceInk/Models/PredefinedModels.swift
View File

@ -1,31 +1,5 @@
import Foundation
struct LocalModel: TranscriptionModel {
let id = UUID()
let name: String
let displayName: String
let size: String
let supportedLanguages: [String: String]
let description: String
let speed: Double
let accuracy: Double
let ramUsage: Double
let hash: String
let provider: ModelProvider = .local
var downloadURL: String {
"https://huggingface.co/ggerganov/whisper.cpp/resolve/main/\(filename)"
}
var filename: String {
"\(name).bin"
}
var isMultilingualModel: Bool {
supportedLanguages.count > 1
}
}
enum PredefinedModels {
static func getLanguageDictionary(isMultilingual: Bool, provider: ModelProvider = .local) -> [String: String] {
if !isMultilingual {
@ -121,8 +95,7 @@ import Foundation
description: "Tiny model, fastest, least accurate",
speed: 0.95,
accuracy: 0.6,
ramUsage: 0.3,
hash: "bd577a113a864445d4c299885e0cb97d4ba92b5f"
ramUsage: 0.3
),
LocalModel(
name: "ggml-tiny.en",
@ -132,8 +105,17 @@ import Foundation
description: "Tiny model optimized for English, fastest, least accurate",
speed: 0.95,
accuracy: 0.65,
ramUsage: 0.3,
hash: "c78c86eb1a8faa21b369bcd33207cc90d64ae9df"
ramUsage: 0.3
),
LocalModel(
name: "ggml-base",
displayName: "Base",
size: "142 MiB",
supportedLanguages: getLanguageDictionary(isMultilingual: true, provider: .local),
description: "Base model, good balance between speed and accuracy, supports multiple languages",
speed: 0.85,
accuracy: 0.72,
ramUsage: 0.5
),
LocalModel(
name: "ggml-base.en",
@ -143,8 +125,7 @@ import Foundation
description: "Base model optimized for English, good balance between speed and accuracy",
speed: 0.85,
accuracy: 0.75,
ramUsage: 0.5,
hash: "137c40403d78fd54d454da0f9bd998f78703390c"
ramUsage: 0.5
),
LocalModel(
name: "ggml-large-v2",
@ -154,8 +135,7 @@ import Foundation
description: "Large model v2, slower than Medium but more accurate",
speed: 0.3,
accuracy: 0.96,
ramUsage: 3.8,
hash: "0f4c8e30f21cf1769f637135f521436792c48186"
ramUsage: 3.8
),
LocalModel(
name: "ggml-large-v3",
@ -165,8 +145,7 @@ import Foundation
description: "Large model v3, very slow but most accurate",
speed: 0.3,
accuracy: 0.98,
ramUsage: 3.9,
hash: "ad82bf6a9043ceed055076d0fd39f5f186ff8062"
ramUsage: 3.9
),
LocalModel(
name: "ggml-large-v3-turbo",
@ -177,8 +156,7 @@ import Foundation
"Large model v3 Turbo, faster than v3 with similar accuracy",
speed: 0.75,
accuracy: 0.97,
ramUsage: 1.8,
hash: "4af2b29d7ec73d781377bfd1758ca957a807e941"
ramUsage: 1.8
),
LocalModel(
name: "ggml-large-v3-turbo-q5_0",
@ -188,8 +166,7 @@ import Foundation
description: "Quantized version of Large v3 Turbo, faster with slightly lower accuracy",
speed: 0.75,
accuracy: 0.95,
ramUsage: 1.0,
hash: "e050f7970618a659205450ad97eb95a18d69c9ee"
ramUsage: 1.0
),
// Cloud Models

25
VoiceInk/Models/TranscriptionModel.swift
View File

@ -95,4 +95,29 @@ struct CustomCloudModel: TranscriptionModel, Codable {
self.isMultilingualModel = isMultilingual
self.supportedLanguages = supportedLanguages ?? PredefinedModels.getLanguageDictionary(isMultilingual: isMultilingual)
}
}
struct LocalModel: TranscriptionModel {
let id = UUID()
let name: String
let displayName: String
let size: String
let supportedLanguages: [String: String]
let description: String
let speed: Double
let accuracy: Double
let ramUsage: Double
let provider: ModelProvider = .local
var downloadURL: String {
"https://huggingface.co/ggerganov/whisper.cpp/resolve/main/\(filename)"
}
var filename: String {
"\(name).bin"
}
var isMultilingualModel: Bool {
supportedLanguages.count > 1
}
}