From 4926ca8471b38da788e77887b58b01cb45ad2687 Mon Sep 17 00:00:00 2001
From: liuzx <liuzx@pcl.ac.cn>
Date: Tue, 22 Feb 2022 10:54:02 +0800
Subject: [PATCH] fix-1541

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

diff --git a/modules/modelarts/modelarts.go b/modules/modelarts/modelarts.go
index 1b7341a56..fed6f8013 100755
--- a/modules/modelarts/modelarts.go
+++ b/modules/modelarts/modelarts.go
@@ -306,7 +306,7 @@ func GenerateNotebook2(ctx *context.Context, jobName, uuid, description, flavor,
 		Type:            models.TypeCloudBrainTwo,
 		Uuid:            uuid,
 		ComputeResource: models.NPUResource,
-		Image: 			 imageName,
+		Image:           imageName,
 	})
 
 	if err != nil {
@@ -594,6 +594,7 @@ func GenerateInferenceJob(ctx *context.Context, req *GenerateInferenceJobReq) (e
 		EngineName:        req.EngineName,
 		LabelName:         req.LabelName,
 		IsLatestVersion:   req.IsLatestVersion,
+		ComputeResource:   models.NPUResource,
 		VersionCount:      req.VersionCount,
 		TotalVersionCount: req.TotalVersionCount,
 		ModelName:         req.ModelName,