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