mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-01-24 14:05:00 +03:00
52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
from dataclasses import dataclass, field
|
|
|
|
from voice_changer.RVC.ModelSlot import ModelSlot
|
|
|
|
|
|
@dataclass
|
|
class RVCSettings:
|
|
gpu: int = 0
|
|
dstId: int = 0
|
|
|
|
f0Detector: str = "harvest" # dio or harvest
|
|
tran: int = 20
|
|
silentThreshold: float = 0.00001
|
|
extraConvertSize: int = 1024 * 32
|
|
clusterInferRatio: float = 0.1
|
|
|
|
framework: str = "PyTorch" # PyTorch or ONNX
|
|
modelSlots: list[ModelSlot] = field(
|
|
default_factory=lambda: [
|
|
ModelSlot(), # 1
|
|
ModelSlot(), # 2
|
|
ModelSlot(), # 3
|
|
ModelSlot(), # 4
|
|
ModelSlot(), # 5
|
|
ModelSlot(), # 6(merged)
|
|
]
|
|
)
|
|
indexRatio: float = 0
|
|
rvcQuality: int = 0
|
|
silenceFront: int = 1 # 0:off, 1:on
|
|
modelSamplingRate: int = 48000
|
|
modelSlotIndex: int = -1
|
|
|
|
speakers: dict[str, int] = field(default_factory=lambda: {})
|
|
isHalf: int = 1 # 0:off, 1:on
|
|
enableDirectML: int = 0 # 0:off, 1:on
|
|
# ↓mutableな物だけ列挙
|
|
intData = [
|
|
"gpu",
|
|
"dstId",
|
|
"tran",
|
|
"extraConvertSize",
|
|
"rvcQuality",
|
|
"modelSamplingRate",
|
|
"silenceFront",
|
|
"modelSlotIndex",
|
|
"isHalf",
|
|
"enableDirectML",
|
|
]
|
|
floatData = ["silentThreshold", "indexRatio"]
|
|
strData = ["framework", "f0Detector"]
|