diff --git a/desc/inference/inference.api b/desc/inference/inference.api index 392a39e4..b80d31aa 100644 --- a/desc/inference/inference.api +++ b/desc/inference/inference.api @@ -38,9 +38,9 @@ type ( TaskName string `form:"taskName"` TaskDesc string `form:"taskDesc"` ModelType string `form:"modelType"` - Instances []*DeployInstance `form:"instances"` + InstanceIds []int64 `form:"instanceIds"` Strategy string `form:"strategy,,optional"` - StaticWeightMap map[string]int32 `form:"staticWeightMap,optional"` + StaticWeightMap map[string]map[string]int32 `form:"staticWeightMap,optional"` Replica int32 `form:"replicas,optional"` } diff --git a/internal/types/types.go b/internal/types/types.go index 6d7ef5c6..2e72d660 100644 --- a/internal/types/types.go +++ b/internal/types/types.go @@ -5935,13 +5935,13 @@ type ModelNamesResp struct { } type ImageInferenceReq struct { - TaskName string `form:"taskName"` - TaskDesc string `form:"taskDesc"` - ModelType string `form:"modelType"` - Instances []*DeployInstance `form:"instances"` - Strategy string `form:"strategy,,optional"` - StaticWeightMap map[string]int32 `form:"staticWeightMap,optional"` - Replica int32 `form:"replicas,optional"` + TaskName string `form:"taskName"` + TaskDesc string `form:"taskDesc"` + ModelType string `form:"modelType"` + InstanceIds []int64 `form:"instanceIds"` + Strategy string `form:"strategy,,optional"` + StaticWeightMap map[string]map[string]int32 `form:"staticWeightMap,optional"` + Replica int32 `form:"replicas,optional"` } type ImageInferenceResp struct {