change structure

This commit is contained in:
wataru 2023-05-04 13:09:13 +09:00
parent b12c042bd8
commit ec75219f18
9 changed files with 5 additions and 4 deletions

View File

@ -21,8 +21,8 @@ else:
sys.path.append("RVC") sys.path.append("RVC")
import json import json
import resampy import resampy
from voice_changer.RVC.MergeModel import merge_model from voice_changer.RVC.modelMerger.MergeModel import merge_model
from voice_changer.RVC.MergeModelRequest import MergeModelRequest from voice_changer.RVC.modelMerger.MergeModelRequest import MergeModelRequest
from voice_changer.RVC.ModelSlotGenerator import generateModelSlot from voice_changer.RVC.ModelSlotGenerator import generateModelSlot
from voice_changer.RVC.RVCSettings import RVCSettings from voice_changer.RVC.RVCSettings import RVCSettings
from voice_changer.RVC.embedder.Embedder import Embedder from voice_changer.RVC.embedder.Embedder import Embedder
@ -393,7 +393,7 @@ class RVC:
print("[Voice Changer] export2onnx, No pyTorch filepath.") print("[Voice Changer] export2onnx, No pyTorch filepath.")
return {"status": "ng", "path": ""} return {"status": "ng", "path": ""}
import voice_changer.RVC.export2onnx as onnxExporter import voice_changer.RVC.onnx.export2onnx as onnxExporter
output_file_simple = onnxExporter.export2onnx(self.settings.gpu, modelSlot) output_file_simple = onnxExporter.export2onnx(self.settings.gpu, modelSlot)
return { return {

View File

@ -1,5 +1,5 @@
from typing import Dict, Any from typing import Dict, Any
from voice_changer.RVC.MergeModelRequest import MergeModelRequest from voice_changer.RVC.modelMerger.MergeModelRequest import MergeModelRequest
from collections import OrderedDict from collections import OrderedDict
import torch import torch
import tqdm import tqdm

View File

@ -38,6 +38,7 @@ def export2onnx(gpu: int, modelSlot: ModelSlot):
"samplingRate": modelSlot.samplingRate, "samplingRate": modelSlot.samplingRate,
"f0": modelSlot.f0, "f0": modelSlot.f0,
"embChannels": modelSlot.embChannels, "embChannels": modelSlot.embChannels,
# ↓EnumEmbedderTypesのままだとシリアライズできないのでテキスト化
"embedder": modelSlot.embedder.value, "embedder": modelSlot.embedder.value,
} }
gpuMomory = DeviceManager.get_instance().getDeviceMemory(gpu) gpuMomory = DeviceManager.get_instance().getDeviceMemory(gpu)