Merge remote-tracking branch 'gitlink/master'

This commit is contained in:
jagger 2025-03-26 20:14:33 +08:00
commit ee2c724d99
1 changed files with 25 additions and 17 deletions

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