added DeployTask apis
Former-commit-id: a9b5c42c5ff1b390c784f82e730e76405319ede0
This commit is contained in:
parent
8c20e3afea
commit
a152ee7576
|
@ -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,
|
||||||
|
|
Loading…
Reference in New Issue