diff --git a/internal/storeLink/modelarts.go b/internal/storeLink/modelarts.go index 1ed8ead7..d4dffa7d 100644 --- a/internal/storeLink/modelarts.go +++ b/internal/storeLink/modelarts.go @@ -170,15 +170,23 @@ func (m *ModelArtsLink) SubmitTask(ctx context.Context, imageId string, cmd stri Value: s[1], }) } - - inputs = append(inputs, &modelarts.InputTraining{ - Name: "input", - AccessMethod: "parameter", - Remote: &modelarts.RemoteTra{ - Obs: &modelarts.ObsTra{ - ObsUrl: datasetsId + "/", - }, - }}) + if len(datasetsId) == 0 { + inputs = append(inputs, &modelarts.InputTraining{ + Name: "input", + AccessMethod: "parameter", + Remote: &modelarts.RemoteTra{ + Obs: &modelarts.ObsTra{ + ObsUrl: datasetsId + "/", + }, + }}) + } else { + inputs = append(inputs, &modelarts.InputTraining{ + Name: "input", + AccessMethod: "parameter", + Remote: &modelarts.RemoteTra{ + Obs: &modelarts.ObsTra{}, + }}) + } /*outputs = append(outputs, &modelarts.OutputTraining{ Name: "output", @@ -908,20 +916,20 @@ func (m *ModelArtsLink) GetImageInferResult(ctx context.Context, url string, fil func (m *ModelArtsLink) CreateInferDeployInstance(ctx context.Context, option *option.InferOption) (string, error) { - err := m.GetModelId(ctx, option) - if err != nil { - return "", err - } + /* err := m.GetModelId(ctx, option) + if err != nil { + return "", err + }*/ - err = m.GetModelStatus(ctx, option) + /*err := m.GetModelStatus(ctx, option) if err != nil { - return "", err - } + return "模型状态查询错误", err + }*/ configParam := &modelarts.ServiceConfig{ Specification: "modelarts.kat1.xlarge", Weight: 100, - ModelId: option.ModelId, + ModelId: option.ModelID, InstanceCount: 1, } var configItems []*modelarts.ServiceConfig