From d3dff0317d69ca3981b0d9989bd7a89f7f64864e Mon Sep 17 00:00:00 2001 From: zouap Date: Tue, 19 Sep 2023 16:25:30 +0800 Subject: [PATCH] =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E5=AE=89=E5=85=A8=E8=AF=84?= =?UTF-8?q?=E6=B5=8B=E7=9B=B8=E5=85=B3Issue=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zouap --- routers/repo/aisafety.go | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go index 1e61b1f266..362f6e7c14 100644 --- a/routers/repo/aisafety.go +++ b/routers/repo/aisafety.go @@ -186,6 +186,9 @@ func DelAiSafetyTask(ctx *context.Context) { } if task.Type == models.TypeCloudBrainOne { cloudbrainTask.DeleteCloudbrainJobStorage(task.JobName, models.TypeCloudBrainOne) + } else { + modelarts.DelTrainJobVersion(task.JobID, strconv.FormatInt(task.VersionID, 10)) + cloudbrainTask.DeleteCloudbrainJobStorage(task.JobName, models.TypeCloudBrainTwo) } err = models.DeleteJob(task) if err != nil { @@ -743,7 +746,7 @@ func createForNPU(ctx *context.Context, jobName string) error { Value: "s3:/" + ckptUrl, }) uuid := srcDataset + ";" + combatDataset - datasUrlList, dataUrl, datasetNames, isMultiDataset, err := getDatasUrlListByUUIDS(uuid) + datasUrlList, dataUrl, datasetNames, _, err := getDatasUrlListByUUIDS(uuid) if err != nil { return err } @@ -753,12 +756,11 @@ func createForNPU(ctx *context.Context, jobName string) error { log.Error("Failed to Marshal: %v", err) return err } - if isMultiDataset { - param = append(param, models.Parameter{ - Label: modelarts.MultiDataUrl, - Value: string(jsondatas), - }) - } + + param = append(param, models.Parameter{ + Label: modelarts.MultiDataUrl, + Value: string(jsondatas), + }) existDeviceTarget := false if len(Params) != 0 { -- 2.34.1