Remove isLargeV3 flag complexity
This commit is contained in:
parent
00e1d5e8da
commit
96a3f35680
@ -27,18 +27,11 @@ import Foundation
|
||||
}
|
||||
|
||||
enum PredefinedModels {
|
||||
static func getLanguageDictionary(isMultilingual: Bool, isLargeV3: Bool = false) -> [String:
|
||||
String]
|
||||
{
|
||||
static func getLanguageDictionary(isMultilingual: Bool) -> [String: String] {
|
||||
if !isMultilingual {
|
||||
return ["en": "English"]
|
||||
} else if isLargeV3 {
|
||||
return allLanguages // Large v3 models support all languages including Cantonese
|
||||
} else {
|
||||
// Create a dictionary without Cantonese for non-Large v3 models
|
||||
var languagesWithoutCantonese = allLanguages
|
||||
languagesWithoutCantonese.removeValue(forKey: "yue")
|
||||
return languagesWithoutCantonese
|
||||
return allLanguages
|
||||
}
|
||||
}
|
||||
|
||||
@ -92,7 +85,7 @@ import Foundation
|
||||
name: "ggml-large-v3",
|
||||
displayName: "Large v3",
|
||||
size: "2.9 GiB",
|
||||
supportedLanguages: getLanguageDictionary(isMultilingual: true, isLargeV3: true),
|
||||
supportedLanguages: getLanguageDictionary(isMultilingual: true),
|
||||
description: "Large model v3, very slow but most accurate",
|
||||
speed: 0.3,
|
||||
accuracy: 0.98,
|
||||
@ -103,7 +96,7 @@ import Foundation
|
||||
name: "ggml-large-v3-turbo",
|
||||
displayName: "Large v3 Turbo",
|
||||
size: "1.5 GiB",
|
||||
supportedLanguages: getLanguageDictionary(isMultilingual: true, isLargeV3: true),
|
||||
supportedLanguages: getLanguageDictionary(isMultilingual: true),
|
||||
description:
|
||||
"Large model v3 Turbo, faster than v3 with similar accuracy",
|
||||
speed: 0.75,
|
||||
@ -115,7 +108,7 @@ import Foundation
|
||||
name: "ggml-large-v3-turbo-q5_0",
|
||||
displayName: "Large v3 Turbo (Quantized)",
|
||||
size: "547 MiB",
|
||||
supportedLanguages: getLanguageDictionary(isMultilingual: true, isLargeV3: true),
|
||||
supportedLanguages: getLanguageDictionary(isMultilingual: true),
|
||||
description: "Quantized version of Large v3 Turbo, faster with slightly lower accuracy",
|
||||
speed: 0.75,
|
||||
accuracy: 0.95,
|
||||
@ -132,7 +125,7 @@ import Foundation
|
||||
speed: 0.65,
|
||||
accuracy: 0.96,
|
||||
isMultilingual: true,
|
||||
supportedLanguages: getLanguageDictionary(isMultilingual: true, isLargeV3: true)
|
||||
supportedLanguages: getLanguageDictionary(isMultilingual: true)
|
||||
),
|
||||
CloudModel(
|
||||
name: "scribe_v1",
|
||||
@ -142,7 +135,7 @@ import Foundation
|
||||
speed: 0.7,
|
||||
accuracy: 0.98,
|
||||
isMultilingual: true,
|
||||
supportedLanguages: getLanguageDictionary(isMultilingual: true, isLargeV3: true)
|
||||
supportedLanguages: getLanguageDictionary(isMultilingual: true)
|
||||
),
|
||||
CloudModel(
|
||||
name: "nova-2",
|
||||
@ -152,7 +145,7 @@ import Foundation
|
||||
speed: 0.9,
|
||||
accuracy: 0.95,
|
||||
isMultilingual: true,
|
||||
supportedLanguages: getLanguageDictionary(isMultilingual: true, isLargeV3: true)
|
||||
supportedLanguages: getLanguageDictionary(isMultilingual: true)
|
||||
),
|
||||
]
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user