From 68d9c7d3b70b3eb3d7cd62d819efcfa903d1be5a Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Thu, 30 Dec 2021 18:30:59 +0800
Subject: [PATCH] mod computeresource

---
 modules/modelarts/modelarts.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/modelarts/modelarts.go b/modules/modelarts/modelarts.go
index 06eac1665..3f7ebfd91 100755
--- a/modules/modelarts/modelarts.go
+++ b/modules/modelarts/modelarts.go
@@ -357,7 +357,7 @@ func GenerateTrainJobVersion(ctx *context.Context, req *GenerateTrainJobReq, job
 		CommitID:          req.CommitID,
 		IsLatestVersion:   req.IsLatestVersion,
 		PreVersionName:    req.PreVersionName,
-		ComputeResource:   models.GPUResource,
+		ComputeResource:   models.NPUResource,
 		EngineID:          req.EngineID,
 		TrainUrl:          req.TrainUrl,
 		BranchName:        req.BranchName,