From 4aa95f8c69b66c1c34e76bc514038afd84e3d33e Mon Sep 17 00:00:00 2001
From: qiwang <1364512070@qq.com>
Date: Thu, 27 Feb 2025 18:52:38 +0800
Subject: [PATCH] fix:update flavor_id in modelarts

---
 internal/storeLink/modelarts.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/internal/storeLink/modelarts.go b/internal/storeLink/modelarts.go
index 806baa04..db507043 100644
--- a/internal/storeLink/modelarts.go
+++ b/internal/storeLink/modelarts.go
@@ -204,7 +204,7 @@ func (m *ModelArtsLink) SubmitTask(ctx context.Context, imageId string, cmd stri
 		},
 		Spec: &modelarts.SpecsC{
 			Resource: &modelarts.ResourceCreateTraining{
-				FlavorId:  "modelarts.kat1.xlarge",
+				FlavorId:  resourceId,
 				NodeCount: 1,
 			},
 		},
@@ -647,7 +647,13 @@ func (m *ModelArtsLink) Execute(ctx context.Context, option *option.AiOption, mo
 			if ascendNum == v.FlavorInfo.Npu.UnitNum {
 				option.ResourceId = v.FlavorId
 				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"
 				break
 			} else if ascendNum >= 5 && ascendNum <= 8 {