-
Notifications
You must be signed in to change notification settings - Fork 2.2k
Expand file tree
/
Copy pathconstants.py
More file actions
32 lines (31 loc) · 993 Bytes
/
Copy pathconstants.py
File metadata and controls
32 lines (31 loc) · 993 Bytes
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
# Define model constants
MODEL_CHECKPOINTS = {
# German vocos checkpoints
"kisz-german-vocos": "hf://aihpi/F5-TTS-German/F5TTS_Base/model_365000.safetensors",
# German bigvgan checkpoints
"kisz-german-bigvgan": "hf://aihpi/F5-TTS-German/F5TTS_Base_bigvgan/model_430000.safetensors",
# English checkpoints
"original-english-vocos": "hf://SWivid/F5-TTS/F5TTS_Base/model_1200000.pt",
"original-english-bigvgan": "hf://SWivid/F5-TTS/F5TTS_Base_bigvgan/model_1250000.pt"
}
MODEL_METADATA = {
# German vocos checkpoints
"kisz-german-vocos": {
"vocoder_name": "vocos",
"language": "de"
},
# German bigvgan checkpoints
"kisz-german-bigvgan": {
"vocoder_name": "bigvgan",
"language": "de"
},
# English checkpoints
"original-english-vocos": {
"vocoder_name": "vocos",
"language": "en"
},
"original-english-bigvgan": {
"vocoder_name": "bigvgan",
"language": "en"
}
}