20 lines
332 B
Python
20 lines
332 B
Python
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class FilePaths:
|
|
configFilename: str
|
|
pyTorchModelFilename: str
|
|
onnxModelFilename: str
|
|
clusterTorchModelFilename: str
|
|
featureFilename: str
|
|
indexFilename: str
|
|
|
|
|
|
@dataclass
|
|
class LoadModelParams:
|
|
slot: int
|
|
isHalf: bool
|
|
files: FilePaths
|
|
params: str
|