routers/repo/modelarts.go

@@ -2,7 +2,6 @@ package repo
 
 import (
 	"archive/zip"
-	ai_task "code.gitea.io/gitea/services/ai_task_service/task"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -16,6 +15,8 @@ import (
 	"time"
 	"unicode/utf8"
 
+	ai_task "code.gitea.io/gitea/services/ai_task_service/task"
+
 	"code.gitea.io/gitea/services/cloudbrain/modelmanage"
 	"code.gitea.io/gitea/services/lock"
 
@@ -2359,7 +2360,7 @@ func InferenceJobCreate(ctx *context.Context, form auth.CreateModelArtsInference
 
 	param = addModelUrlParam(param, form.PreTrainModelUrl, form.CkptName)
 
-	datasUrlList, dataUrl, datasetNames, isMultiDataset, err := getDatasUrlListByUUIDS(uuid)
+	datasUrlList, dataUrl, datasetNames, _, err := getDatasUrlListByUUIDS(uuid)
 	if err != nil {
 		inferenceJobErrorNewDataPrepare(ctx, form)
 		ctx.RenderWithErr(err.Error(), tplModelArtsInferenceJobNew, &form)
@@ -2373,12 +2374,10 @@ func InferenceJobCreate(ctx *context.Context, form auth.CreateModelArtsInference
 		ctx.RenderWithErr("json error:"+err.Error(), tplModelArtsInferenceJobNew, &form)
 		return
 	}
-	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 {