Browse Source

Merge remote-tracking branch 'gitlink/master'

pull/464/head
jagger 7 months ago
parent
commit
ee2c724d99
1 changed files with 25 additions and 17 deletions
  1. +25
    -17
      internal/storeLink/modelarts.go

+ 25
- 17
internal/storeLink/modelarts.go View File

@@ -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


Loading…
Cancel
Save