mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-02-03 08:43:57 +03:00
56 lines
1.0 KiB
Python
56 lines
1.0 KiB
Python
from dataclasses import dataclass
|
|
from typing import Any
|
|
|
|
from const import VoiceChangerType
|
|
from typing import Literal, TypeAlias
|
|
|
|
|
|
@dataclass
|
|
class FilePaths:
|
|
configFilename: str | None
|
|
pyTorchModelFilename: str | None
|
|
onnxModelFilename: str | None
|
|
clusterTorchModelFilename: str | None
|
|
featureFilename: str | None
|
|
indexFilename: str | None
|
|
|
|
|
|
@dataclass
|
|
class LoadModelParams:
|
|
slot: int
|
|
isHalf: bool
|
|
params: Any
|
|
|
|
|
|
LoadModelParamFileKind: TypeAlias = Literal[
|
|
"mmvcv13Config",
|
|
"mmvcv13Model",
|
|
"mmvcv15Config",
|
|
"mmvcv15Model",
|
|
"soVitsSvc40Config",
|
|
"soVitsSvc40Model",
|
|
"soVitsSvc40Cluster",
|
|
"rvcModel",
|
|
"rvcIndex",
|
|
"ddspSvcModel",
|
|
"ddspSvcModelConfig",
|
|
"ddspSvcDiffusion",
|
|
"ddspSvcDiffusionConfig",
|
|
]
|
|
|
|
|
|
@dataclass
|
|
class LoadModelParamFile:
|
|
name: str
|
|
kind: LoadModelParamFileKind
|
|
dir: str
|
|
|
|
|
|
@dataclass
|
|
class LoadModelParams2:
|
|
voiceChangerType: VoiceChangerType
|
|
slot: int
|
|
isSampleMode: bool
|
|
sampleId: str
|
|
files: list[LoadModelParamFile]
|