forked from JointCloud/pcm-coordinator
39 lines
1.1 KiB
Go
39 lines
1.1 KiB
Go
package option
|
|
|
|
type InferOption struct {
|
|
TaskName string `json:"taskName"`
|
|
TaskDesc string `json:"taskDesc"`
|
|
ModelName string `json:"modelName"`
|
|
ModelType string `json:"modelType"`
|
|
AdapterId string `json:"adapterId"`
|
|
AiClusterIds []string `json:"aiClusterIds"`
|
|
ResourceType string `json:"resourceType"`
|
|
ComputeCard string `json:"card"`
|
|
Strategy string `json:"strategy"`
|
|
StaticWeightMap map[string]int32 `json:"staticWeightMap,optional"`
|
|
Params []string `json:"params,optional"`
|
|
Envs []string `json:"envs,optional"`
|
|
Cmd string `json:"cmd,optional"`
|
|
Replica int32 `json:"replicas,optional"`
|
|
|
|
ResourceId string
|
|
AlgorithmId string
|
|
ImageId string
|
|
ModelID string
|
|
|
|
Output string
|
|
|
|
ResourcesRequired []map[string]interface{}
|
|
|
|
//modelarts
|
|
InferType string
|
|
Specification string
|
|
Weight int32
|
|
ModelId string
|
|
InstanceCount int32
|
|
}
|
|
|
|
func (a InferOption) GetOptionType() string {
|
|
return AI_INFER
|
|
}
|