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:
|
2023-05-09 01:01:20 +09:00
|
|
|
modelFile: str = ""
|
2023-04-28 07:36:08 +09:00
|
|
|
indexFile: str = ""
|
2023-05-17 12:37:35 +09:00
|
|
|
defaultTune: int = 0
|
|
|
|
defaultIndexRatio: int = 1
|
2023-06-01 13:28:45 +09:00
|
|
|
defaultProtect: float = 0.5
|
2023-05-02 20:57:12 +09:00
|
|
|
isONNX: bool = False
|
2023-05-31 02:26:16 +09:00
|
|
|
modelType: str = EnumInferenceTypes.pyTorchRVC.value
|
2023-04-28 07:36:08 +09:00
|
|
|
samplingRate: int = -1
|
|
|
|
f0: bool = True
|
|
|
|
embChannels: int = 256
|
2023-05-24 20:49:24 +09:00
|
|
|
embOutputLayer: int = 9
|
2023-05-24 16:56:23 +09:00
|
|
|
useFinalProj: bool = True
|
2023-04-28 07:36:08 +09:00
|
|
|
deprecated: bool = False
|
2023-05-31 02:26:16 +09:00
|
|
|
embedder: str = EnumEmbedderTypes.hubert.value
|
2023-05-16 10:38:23 +09:00
|
|
|
|
|
|
|
name: str = ""
|
|
|
|
description: str = ""
|
|
|
|
credit: str = ""
|
|
|
|
termsOfUseUrl: str = ""
|
2023-05-31 02:26:16 +09:00
|
|
|
sampleId: str = ""
|