File size: 2,735 Bytes
1d74c6a |
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 |
from voice_changer.RVC.RVCModelSlotGenerator import RVCModelSlotGenerator
from voice_changer.VoiceChangerParamsManager import VoiceChangerParamsManager
from voice_changer.utils.LoadModelParams import LoadModelParamFile, LoadModelParams
from voice_changer.utils.VoiceChangerParams import VoiceChangerParams
model_dir1 = "model_dir/"+model_slot+"/"
is_pth = True # Set this to True if you want to search for .pth files, or False for .onnx files
file_extension = ".pth" if is_pth else ".onnx"
# pth_files = [f for f in os.listdir(model_dir1) if f.endswith(file_extension)]
pth_files = [f for f in os.listdir(model_dir1) if f.endswith(".pth") or f.endswith(".onnx")]
print(pth_files)
index_files = [f for f in os.listdir(model_dir1) if f.endswith(".index")]
if pth_files:
model_name = pth_files[0].replace(".pth", "")
else:
model_name = "Null"
if index_files:
index_name = index_files[0].replace(".index", "")
else:
index_name = ""
original_string = str(pth_files)
string_pth_files = original_string[2:-2]
print("IM A STRING"+original_string)
print(model_name)
voiceChangerParams = VoiceChangerParams(
model_dir="./model_dir/"+model_slot,
content_vec_500="",
content_vec_500_onnx="",
content_vec_500_onnx_on="",
hubert_base="",
hubert_base_jp="",
hubert_soft="",
nsf_hifigan="",
crepe_onnx_full="",
crepe_onnx_tiny="",
rmvpe="",
rmvpe_onnx="",
sample_mode=""
)
vcparams = VoiceChangerParamsManager.get_instance()
vcparams.setParams(voiceChangerParams)
file = LoadModelParamFile(
name=string_pth_files,
kind="rvcModel",
dir="",
)
loadParam = LoadModelParams(
voiceChangerType="RVC",
files=[file],
slot="",
isSampleMode=False,
sampleId="",
params={},
)
slotInfo = RVCModelSlotGenerator.loadModel(loadParam)
print(slotInfo.samplingRate)
#---------------Make the Json File-----------
params_content = {
"slotIndex": -1,
"voiceChangerType": "RVC",
"name": model_name,
"description": "",
"credit": "",
"termsOfUseUrl": "",
"iconFile": iconFile,
"speakers": {
"0": "target"
},
"modelFile": string_pth_files,
"indexFile": f"{index_name}.index",
"defaultTune": Tune,
"defaultIndexRatio": Index,
"defaultProtect": 0.5,
"isONNX": False,
"modelType": "pyTorchRVCv2",
"samplingRate": slotInfo.samplingRate,
"f0": True,
"embChannels": 768,
"embOutputLayer": 12,
"useFinalProj": False,
"deprecated": False,
"embedder": "hubert_base",
"sampleId": ""
}
# Write the content to params.json
with open(f"{model_dir1}/params.json", "w") as param_file:
json.dump(params_content, param_file) |