floraachy
/
pcm-coordinator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated imageinference api 

Former-commit-id: f8b560cea1c26f05353800af2e2c0f16a885cc85
This commit is contained in:
tzwang 2024-06-20 17:16:06 +08:00
parent 99d96327ac
commit bee590f969
2 changed files with 28 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
api/desc/inference/inference.api
View File

@ -1,26 +1,23 @@
syntax = "v1"
type (
InferOption {
TaskName string `json:"taskName"`
TaskDesc string `json:"taskDesc"`
ModelName string `json:"modelName"`
ModelType string `json:"modelType"`
AdapterId string `json:"adapterId"`
AiClusterIds []string `json:"aiClusterIds,optional"`
ResourceType string `json:"resourceType,optional"`
ComputeCard string `json:"card,optional"`
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"`
}
/******************image inference*************************/
ImageInferenceReq {
InferOption *InferOption `json:"inferOption,optional"`
TaskName string `form:"taskName"`
TaskDesc string `form:"taskDesc"`
ModelName string `form:"modelName"`
ModelType string `form:"modelType"`
AdapterId string `form:"adapterId"`
AiClusterIds []string `form:"aiClusterIds,optional"`
ResourceType string `form:"resourceType,optional"`
ComputeCard string `form:"card,optional"`
Strategy string `form:"strategy"`
StaticWeightMap map[string]int32 `form:"staticWeightMap,optional"`
Params []string `form:"params,optional"`
Envs []string `form:"envs,optional"`
Cmd string `form:"cmd,optional"`
Replica int32 `form:"replicas,optional"`
}
ImageInferenceResp {

32
api/internal/types/types.go
View File

@ -5879,25 +5879,21 @@ type Category struct {
Name string `json:"name"`
}
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,optional"`
ResourceType string `json:"resourceType,optional"`
ComputeCard string `json:"card,optional"`
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"`
}
type ImageInferenceReq struct {
InferOption *InferOption `json:"inferOption,optional"`
TaskName string `form:"taskName"`
TaskDesc string `form:"taskDesc"`
ModelName string `form:"modelName"`
ModelType string `form:"modelType"`
AdapterId string `form:"adapterId"`
AiClusterIds []string `form:"aiClusterIds,optional"`
ResourceType string `form:"resourceType,optional"`
ComputeCard string `form:"card,optional"`
Strategy string `form:"strategy"`
StaticWeightMap map[string]int32 `form:"staticWeightMap,optional"`
Params []string `form:"params,optional"`
Envs []string `form:"envs,optional"`
Cmd string `form:"cmd,optional"`
Replica int32 `form:"replicas,optional"`
}
type ImageInferenceResp struct {