28 lines
669 B
Python
Raw Normal View History

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 = ""