mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-01-23 21:45:00 +03:00
62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
from dataclasses import dataclass, field
|
|
from typing import TypeAlias, Union, Any
|
|
from const import VoiceChangerType
|
|
|
|
|
|
@dataclass
|
|
class ModelSample:
|
|
id: str = ""
|
|
voiceChangerType: VoiceChangerType | None = None
|
|
|
|
|
|
@dataclass
|
|
class RVCModelSample(ModelSample):
|
|
id: str = ""
|
|
voiceChangerType: VoiceChangerType = "RVC"
|
|
lang: str = ""
|
|
tag: list[str] = field(default_factory=lambda: [])
|
|
name: str = ""
|
|
modelUrl: str = ""
|
|
indexUrl: str = ""
|
|
termsOfUseUrl: str = ""
|
|
icon: str = ""
|
|
credit: str = ""
|
|
description: str = ""
|
|
|
|
sampleRate: int = 48000
|
|
modelType: str = ""
|
|
f0: bool = True
|
|
|
|
|
|
@dataclass
|
|
class DiffusionSVCModelSample(ModelSample):
|
|
id: str = ""
|
|
voiceChangerType: VoiceChangerType = "Diffusion-SVC"
|
|
lang: str = ""
|
|
tag: list[str] = field(default_factory=lambda: [])
|
|
name: str = ""
|
|
modelUrl: str = ""
|
|
termsOfUseUrl: str = ""
|
|
icon: str = ""
|
|
credit: str = ""
|
|
description: str = ""
|
|
|
|
sampleRate: int = 48000
|
|
modelType: str = ""
|
|
f0: bool = True
|
|
numOfDiffLayers: int = 20
|
|
numOfNativeLayers: int = 3
|
|
maxKStep: int = 50
|
|
|
|
|
|
ModelSamples: TypeAlias = Union[ModelSample, RVCModelSample, DiffusionSVCModelSample]
|
|
|
|
|
|
def generateModelSample(params: Any) -> ModelSamples:
|
|
if params["voiceChangerType"] == "RVC":
|
|
return RVCModelSample(**{k: v for k, v in params.items() if k in RVCModelSample.__annotations__})
|
|
elif params["voiceChangerType"] == "Diffusion-SVC":
|
|
return DiffusionSVCModelSample(**{k: v for k, v in params.items() if k in DiffusionSVCModelSample.__annotations__})
|
|
else:
|
|
return ModelSample(**{k: v for k, v in params.items() if k in ModelSample.__annotations__})
|