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
}
