| @@ -204,7 +204,7 @@ func (m *ModelArtsLink) SubmitTask(ctx context.Context, imageId string, cmd stri | |||||
| }, | }, | ||||
| Spec: &modelarts.SpecsC{ | Spec: &modelarts.SpecsC{ | ||||
| Resource: &modelarts.ResourceCreateTraining{ | Resource: &modelarts.ResourceCreateTraining{ | ||||
| FlavorId: "modelarts.kat1.xlarge", | |||||
| FlavorId: resourceId, | |||||
| NodeCount: 1, | NodeCount: 1, | ||||
| }, | }, | ||||
| }, | }, | ||||
| @@ -647,7 +647,13 @@ func (m *ModelArtsLink) Execute(ctx context.Context, option *option.AiOption, mo | |||||
| if ascendNum == v.FlavorInfo.Npu.UnitNum { | if ascendNum == v.FlavorInfo.Npu.UnitNum { | ||||
| option.ResourceId = v.FlavorId | option.ResourceId = v.FlavorId | ||||
| break | break | ||||
| } else if ascendNum == 3 { | |||||
| } else if ascendNum <= 1 { | |||||
| option.ResourceId = "modelarts.kat1.xlarge" | |||||
| break | |||||
| } else if ascendNum == 2 { | |||||
| option.ResourceId = "modelarts.kat1.2xlarge" | |||||
| break | |||||
| } else if ascendNum > 2 && ascendNum <= 4 { | |||||
| option.ResourceId = "modelarts.kat1.4xlarge" | option.ResourceId = "modelarts.kat1.4xlarge" | ||||
| break | break | ||||
| } else if ascendNum >= 5 && ascendNum <= 8 { | } else if ascendNum >= 5 && ascendNum <= 8 { | ||||