2023-05-02 20:57:12 +09:00
|
|
|
from const import EnumInferenceTypes, EnumEmbedderTypes
|
|
|
|
|
|
2023-04-28 07:36:08 +09:00
|
|
|
from dataclasses import dataclass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@dataclass
|
|
|
|
|
class ModelSlot:
|
|
|
|
|
pyTorchModelFile: str = ""
|
|
|
|
|
onnxModelFile: str = ""
|
|
|
|
|
featureFile: str = ""
|
|
|
|
|
indexFile: str = ""
|
|
|
|
|
defaultTrans: int = 0
|
2023-05-02 20:57:12 +09:00
|
|
|
isONNX: bool = False
|
|
|
|
|
modelType: EnumInferenceTypes = EnumInferenceTypes.pyTorchRVC
|
2023-04-28 07:36:08 +09:00
|
|
|
samplingRate: int = -1
|
|
|
|
|
f0: bool = True
|
|
|
|
|
embChannels: int = 256
|
|
|
|
|
deprecated: bool = False
|
2023-05-02 20:57:12 +09:00
|
|
|
embedder: EnumEmbedderTypes = EnumEmbedderTypes.hubert
|
