Applio_V3 / rvc /lib /tools /pretrained_selector.py
freshyfresco's picture
Upload 125 files (#2)
938411e verified
raw
history blame
2.33 kB
def pretrained_selector(pitch_guidance):
if pitch_guidance:
return {
"v1": {
"32000": (
"rvc/pretraineds/pretrained_v1/f0G32k.pth",
"rvc/pretraineds/pretrained_v1/f0D32k.pth",
),
"40000": (
"rvc/pretraineds/pretrained_v1/f0G40k.pth",
"rvc/pretraineds/pretrained_v1/f0D40k.pth",
),
"48000": (
"rvc/pretraineds/pretrained_v1/f0G48k.pth",
"rvc/pretraineds/pretrained_v1/f0D48k.pth",
),
},
"v2": {
"32000": (
"rvc/pretraineds/pretrained_v2/f0G32k.pth",
"rvc/pretraineds/pretrained_v2/f0D32k.pth",
),
"40000": (
"rvc/pretraineds/pretrained_v2/f0G40k.pth",
"rvc/pretraineds/pretrained_v2/f0D40k.pth",
),
"48000": (
"rvc/pretraineds/pretrained_v2/f0G48k.pth",
"rvc/pretraineds/pretrained_v2/f0D48k.pth",
),
},
}
else:
return {
"v1": {
"32000": (
"rvc/pretraineds/pretrained_v1/G32k.pth",
"rvc/pretraineds/pretrained_v1/D32k.pth",
),
"40000": (
"rvc/pretraineds/pretrained_v1/G40k.pth",
"rvc/pretraineds/pretrained_v1/D40k.pth",
),
"48000": (
"rvc/pretraineds/pretrained_v1/G48k.pth",
"rvc/pretraineds/pretrained_v1/D48k.pth",
),
},
"v2": {
"32000": (
"rvc/pretraineds/pretrained_v2/G32k.pth",
"rvc/pretraineds/pretrained_v2/D32k.pth",
),
"40000": (
"rvc/pretraineds/pretrained_v2/G40k.pth",
"rvc/pretraineds/pretrained_v2/D40k.pth",
),
"48000": (
"rvc/pretraineds/pretrained_v2/G48k.pth",
"rvc/pretraineds/pretrained_v2/D48k.pth",
),
},
}