PublicMirrors/voice-changer
1
0
Fork 0
mirror of https://github.com/w-okada/voice-changer.git synced 2025-02-03 00:33:57 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

WIP:common sample

Browse Source
This commit is contained in:
wataru 2023-06-16 16:53:33 +09:00
parent 99cf22041f
commit 87de3dc10f
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
server/voice_changer/RVC/RVC.py
View File

@ -5,7 +5,7 @@ from typing import cast
import numpy as np
import torch
import torchaudio
from ModelSample import getModelSamples
from utils.downloader.SampleDownloader import getSampleInfos
from voice_changer.RVC.ModelSlot import ModelSlot
from voice_changer.RVC.SampleDownloader import downloadModelFiles
@ -44,7 +44,6 @@ from Exceptions import DeviceCannotSupportHalfPrecisionException, NoModeLoadedEx
from const import (
RVC_MODEL_DIRNAME,
UPLOAD_DIR,
getRVCSampleJsonAndModelIds,
)
import shutil
import json
@ -72,15 +71,15 @@ class RVC:
print("[Voice Changer] RVC initialization: ", params)
# サンプルカタログ作成
sampleJsons: list[str] = []
sampleJsonUrls, _sampleModels = getRVCSampleJsonAndModelIds(params.rvc_sample_mode)
for url in sampleJsonUrls:
filename = os.path.basename(url)
sampleJsons.append(filename)
sampleModels = getModelSamples(sampleJsons, "RVC")
if sampleModels is not None:
self.settings.sampleModels = sampleModels
# sampleJsons: list[str] = []
samples = getSampleInfos(params.sample_mode)
# for url in sampleJsonUrls:
# filename = os.path.basename(url)
# sampleJsons.append(filename)
# sampleModels = getModelSamples(sampleJsons, "RVC")
# if sampleModels is not None:
# self.settings.sampleModels = sampleModels
self.settings.sampleModels = samples
# 起動時にスロットにモデルがある場合はロードしておく
if len(self.settings.modelSlots) > 0:
for i, slot in enumerate(self.settings.modelSlots):