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")
import json
import resampy
from voice_changer.RVC.MergeModel import merge_model
from voice_changer.RVC.MergeModelRequest import MergeModelRequest
from voice_changer.RVC.modelMerger.MergeModel import merge_model
from voice_changer.RVC.modelMerger.MergeModelRequest import MergeModelRequest
from voice_changer.RVC.ModelSlotGenerator import generateModelSlot
from voice_changer.RVC.RVCSettings import RVCSettings
from voice_changer.RVC.embedder.Embedder import Embedder
@ -393,7 +393,7 @@ class RVC:
print("[Voice Changer] export2onnx, No pyTorch filepath.")
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)
return {

View File

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

View File

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