Spaces:
Runtime error
Runtime error
File size: 2,508 Bytes
a8c39f5 |
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 |
def pretrained_selector(pitch_guidance):
if pitch_guidance == True:
return {
"v1": {
32000: (
"rvc/models/pretraineds/pretrained_v1/f0G32k.pth",
"rvc/models/pretraineds/pretrained_v1/f0D32k.pth",
),
40000: (
"rvc/models/pretraineds/pretrained_v1/f0G40k.pth",
"rvc/models/pretraineds/pretrained_v1/f0D40k.pth",
),
48000: (
"rvc/models/pretraineds/pretrained_v1/f0G48k.pth",
"rvc/models/pretraineds/pretrained_v1/f0D48k.pth",
),
},
"v2": {
32000: (
"rvc/models/pretraineds/pretrained_v2/f0G32k.pth",
"rvc/models/pretraineds/pretrained_v2/f0D32k.pth",
),
40000: (
"rvc/models/pretraineds/pretrained_v2/f0G40k.pth",
"rvc/models/pretraineds/pretrained_v2/f0D40k.pth",
),
48000: (
"rvc/models/pretraineds/pretrained_v2/f0G48k.pth",
"rvc/models/pretraineds/pretrained_v2/f0D48k.pth",
),
},
}
elif pitch_guidance == False:
return {
"v1": {
32000: (
"rvc/models/pretraineds/pretrained_v1/G32k.pth",
"rvc/models/pretraineds/pretrained_v1/D32k.pth",
),
40000: (
"rvc/models/pretraineds/pretrained_v1/G40k.pth",
"rvc/models/pretraineds/pretrained_v1/D40k.pth",
),
48000: (
"rvc/models/pretraineds/pretrained_v1/G48k.pth",
"rvc/models/pretraineds/pretrained_v1/D48k.pth",
),
},
"v2": {
32000: (
"rvc/models/pretraineds/pretrained_v2/G32k.pth",
"rvc/models/pretraineds/pretrained_v2/D32k.pth",
),
40000: (
"rvc/models/pretraineds/pretrained_v2/G40k.pth",
"rvc/models/pretraineds/pretrained_v2/D40k.pth",
),
48000: (
"rvc/models/pretraineds/pretrained_v2/G48k.pth",
"rvc/models/pretraineds/pretrained_v2/D48k.pth",
),
},
}
|