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,