|
|
@@ -2594,7 +2594,7 @@ func inferenceJobErrorNewDataPrepare(ctx *context.Context, form auth.CreateModel |
|
|
|
waitCount := cloudbrain.GetWaitingCloudbrainCount(models.TypeCloudBrainTwo, "") |
|
|
|
ctx.Data["WaitCount"] = waitCount |
|
|
|
NotStopTaskCount, _ := cloudbrainTask.GetNotFinalStatusTaskCount(ctx.User.ID, models.TypeCloudBrainTwo, string(models.JobTypeInference)) |
|
|
|
ctx.Data["NotStopTaskCount"] = NotStopTaskCount |
|
|
|
ctx.Data["NotStopTaskCount"] = NotStopTaskCount |
|
|
|
return nil |
|
|
|
} |
|
|
|
func InferenceJobShow(ctx *context.Context) { |
|
|
@@ -2674,6 +2674,8 @@ func MultiModelDownload(ctx *context.Context) { |
|
|
|
|
|
|
|
if task.ComputeResource == models.NPUResource { |
|
|
|
path := strings.TrimPrefix(path.Join(setting.TrainJobModelPath, task.JobName, setting.OutPutPath, versionName, parentDir), "/") |
|
|
|
path = strings.TrimSuffix(path, "/") |
|
|
|
path += "/" |
|
|
|
allFile, err := storage.GetAllObjectByBucketAndPrefix(setting.Bucket, path) |
|
|
|
if err == nil { |
|
|
|
returnFileName := task.DisplayJobName + ".zip" |
|
|
|