1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
| package model
|
| import (
| "fmt"
| "sort"
| )
|
| const (
| BackendSenseVoice = "sensevoice"
| )
|
| var modelProfiles = map[string]ModelProfile{
| DefaultModelID: {
| ID: DefaultModelID,
| DisplayName: "SenseVoice",
| BackendKind: BackendSenseVoice,
| Tier: "default",
| SupportedLanguageIDs: []string{
| "zh-CN",
| "zh-Hans",
| "zh-Hant",
| "yue",
| "en",
| "ja",
| "ko",
| },
| RecommendedFor: []string{"zh-CN"},
| Description: "轻量中文模型，适合中文和中英混输。",
| InstallDirName: "sensevoice-zh",
| LegacyDirNames: []string{"sensevoice"},
| RequiredFiles: []RequiredFileRule{
| {
| Role: "model",
| AnyOf: []string{"model.int8.onnx", "model.onnx"},
| Required: true,
| },
| {
| Role: "tokens",
| AllOf: []string{"tokens.txt"},
| Required: true,
| },
| },
| ProviderOrder: []string{"coreml", "cpu"},
| NumThreads: 4,
| },
| }
|
| var languageProfiles = map[string]LanguageProfile{
| DefaultLanguageID: {
| ID: DefaultLanguageID,
| DisplayName: "Chinese",
| NativeName: "中文",
| UILocale: "zh",
| SystemMatchers: []string{"zh", "zh-CN", "zh-Hans", "zh-Hant"},
| DefaultModelID: DefaultModelID,
| },
| EnglishLanguageID: {
| ID: EnglishLanguageID,
| DisplayName: "English",
| NativeName: "English",
| UILocale: "en",
| SystemMatchers: []string{"en", "en-US", "en-GB", "en-AU", "en-CA"},
| DefaultModelID: DefaultModelID,
| },
| }
|
| func GetModelProfile(id string) (ModelProfile, error) {
| if id == "" {
| id = DefaultModelID
| }
| profile, ok := modelProfiles[id]
| if !ok {
| return ModelProfile{}, fmt.Errorf("unknown model profile: %s", id)
| }
| return profile, nil
| }
|
| func DefaultModelProfile() ModelProfile {
| return modelProfiles[DefaultModelID]
| }
|
| func ListModelProfiles() []ModelProfile {
| profiles := make([]ModelProfile, 0, len(modelProfiles))
| for _, profile := range modelProfiles {
| profiles = append(profiles, profile)
| }
| return profiles
| }
|
| func NormalizeModelID(id string) string {
| if _, err := GetModelProfile(id); err != nil {
| return DefaultModelID
| }
| return id
| }
|
| func GetLanguageProfile(id string) (LanguageProfile, error) {
| profile, ok := languageProfiles[id]
| if !ok {
| return LanguageProfile{}, fmt.Errorf("unknown language profile: %s", id)
| }
| return profile, nil
| }
|
| func ListLanguageProfiles() []LanguageProfile {
| profiles := make([]LanguageProfile, 0, len(languageProfiles))
| for _, profile := range languageProfiles {
| profiles = append(profiles, profile)
| }
| sort.Slice(profiles, func(i, j int) bool {
| if profiles[i].ID == DefaultLanguageID {
| return true
| }
| if profiles[j].ID == DefaultLanguageID {
| return false
| }
| return profiles[i].ID < profiles[j].ID
| })
| return profiles
| }
|
| func NormalizeLanguageID(id string) string {
| if _, err := GetLanguageProfile(id); err != nil {
| return DefaultLanguageID
| }
| return id
| }
|
|
