mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-02-03 00:33:57 +03:00
change structure
This commit is contained in:
parent
b12c042bd8
commit
ec75219f18
@ -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 {
|
||||
|
@ -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
|
@ -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)
|
Loading…
Reference in New Issue
Block a user