Browse Source

fix:update flavor_id in modelarts

pull/425/head
qiwang 8 months ago
parent
commit
4aa95f8c69
1 changed files with 8 additions and 2 deletions
  1. +8
    -2
      internal/storeLink/modelarts.go

+ 8
- 2
internal/storeLink/modelarts.go View File

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


Loading…
Cancel
Save