voice-changer/server/voice_changer/RVC/ModelSlot.py
2023-05-16 10:38:23 +09:00

26 lines
632 B
Python

from const import EnumInferenceTypes, EnumEmbedderTypes
from dataclasses import dataclass
@dataclass
class ModelSlot:
# pyTorchModelFile: str = ""
# onnxModelFile: str = ""
modelFile: str = ""
featureFile: str = ""
indexFile: str = ""
defaultTrans: int = 0
isONNX: bool = False
modelType: EnumInferenceTypes = EnumInferenceTypes.pyTorchRVC
samplingRate: int = -1
f0: bool = True
embChannels: int = 256
deprecated: bool = False
embedder: EnumEmbedderTypes = EnumEmbedderTypes.hubert
name: str = ""
description: str = ""
credit: str = ""
termsOfUseUrl: str = ""