|
|
@@ -534,12 +534,14 @@ func DeleteModelConvert(ctx *context.Context) { |
|
|
|
|
|
|
|
func deleteCloudBrainTask(task *models.AiModelConvert) { |
|
|
|
if task.IsGpuTrainTask() { |
|
|
|
log.Info("delete cloudbrain one resource.") |
|
|
|
dirPath := setting.CBCodePathPrefix + task.ID + "/" |
|
|
|
err := storage.Attachments.DeleteDir(dirPath) |
|
|
|
if err != nil { |
|
|
|
log.Error("DeleteDir(%s) failed:%v", dirPath, err) |
|
|
|
} |
|
|
|
} else { |
|
|
|
log.Info("delete cloudbrain two resource.") |
|
|
|
_, err := modelarts.DelTrainJob(task.CloudBrainTaskId) |
|
|
|
if err != nil { |
|
|
|
log.Error("DelTrainJob(%s) failed:%v", task.CloudBrainTaskId, err.Error()) |
|
|
|