21 lines
397 B
Python
21 lines
397 B
Python
from dataclasses import dataclass
|
|
from typing import Any
|
|
|
|
|
|
@dataclass
|
|
class FilePaths:
|
|
configFilename: str | None
|
|
pyTorchModelFilename: str | None
|
|
onnxModelFilename: str | None
|
|
clusterTorchModelFilename: str | None
|
|
featureFilename: str | None
|
|
indexFilename: str | None
|
|
|
|
|
|
@dataclass
|
|
class LoadModelParams:
|
|
slot: int
|
|
isHalf: bool
|
|
files: FilePaths
|
|
params: Any
|