| @@ -120,4 +120,28 @@ type ( | |||||
| Running int32 `json:"running"` | Running int32 `json:"running"` | ||||
| Total int32 `json:"total"` | Total int32 `json:"total"` | ||||
| } | } | ||||
| StartAllByDeployTaskIdReq { | |||||
| Id string `form:"id"` | |||||
| } | |||||
| StartAllByDeployTaskIdResp { | |||||
| } | |||||
| StopAllByDeployTaskIdReq { | |||||
| Id string `form:"id"` | |||||
| } | |||||
| StopAllByDeployTaskIdResp { | |||||
| } | |||||
| GetDeployTasksReq { | |||||
| PageInfo | |||||
| } | |||||
| GetDeployTasksResp { | |||||
| PageResult | |||||
| } | |||||
| ) | ) | ||||
| @@ -959,6 +959,15 @@ service pcm { | |||||
| @handler InferenceTaskStatHandler | @handler InferenceTaskStatHandler | ||||
| get /inference/taskStat (InferenceTaskStatReq) returns (InferenceTaskStatResp) | get /inference/taskStat (InferenceTaskStatReq) returns (InferenceTaskStatResp) | ||||
| @handler StartAllByDeployTaskId | |||||
| post /inference/startAllById (StartAllByDeployTaskIdReq) returns (StartAllByDeployTaskIdResp) | |||||
| @handler StopAllByDeployTaskId | |||||
| post /inference/stopAllById (StopAllByDeployTaskIdReq) returns (StopAllByDeployTaskIdResp) | |||||
| @handler GetDeployTasks | |||||
| get /inference/getDeployTasks (GetDeployTasksReq) returns (GetDeployTasksResp) | |||||
| } | } | ||||
| @server( | @server( | ||||
| @@ -3,7 +3,7 @@ package imageInference | |||||
| import "gitlink.org.cn/JointCloud/pcm-coordinator/internal/scheduler/service/inference" | import "gitlink.org.cn/JointCloud/pcm-coordinator/internal/scheduler/service/inference" | ||||
| const ( | const ( | ||||
| IMAGETOTEXT = "image-to-text" | |||||
| IMAGETOTEXT = "image" | |||||
| IMAGETOTEXT_AiTYPE = "13" | IMAGETOTEXT_AiTYPE = "13" | ||||
| ) | ) | ||||
| @@ -81,6 +81,7 @@ var ( | |||||
| ModelTypeMap = map[string][]string{ | ModelTypeMap = map[string][]string{ | ||||
| "image_classification": {"imagenet_resnet50"}, | "image_classification": {"imagenet_resnet50"}, | ||||
| "text_to_text": {"chatGLM_6B"}, | "text_to_text": {"chatGLM_6B"}, | ||||
| "image_to_text": {"blip-image-captioning-base"}, | |||||
| } | } | ||||
| AITYPE = map[string]string{ | AITYPE = map[string]string{ | ||||
| "1": OCTOPUS, | "1": OCTOPUS, | ||||