routers/repo/aisafety.go

@@ -2,8 +2,6 @@ package repo
 
 import (
 	"bufio"
-	"code.gitea.io/gitea/services/cloudbrain/cloudbrainTask"
-	"code.gitea.io/gitea/services/lock"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -14,6 +12,9 @@ import (
 	"strconv"
 	"strings"
 
+	"code.gitea.io/gitea/services/cloudbrain/cloudbrainTask"
+	"code.gitea.io/gitea/services/lock"
+
 	cloudbrainService "code.gitea.io/gitea/services/cloudbrain"
 
 	"code.gitea.io/gitea/models"
@@ -669,7 +670,7 @@ func createForNPU(ctx *context.Context, jobName string) error {
 	poolID := ctx.Query("pool_id")
 	repo := ctx.Repo.Repository
 
-	trainUrl := ctx.Query("train_url")
+	trainUrl := ctx.Query("pre_train_model_url")
 	modelName := ctx.Query("model_name")
 	modelVersion := ctx.Query("model_version")
 	ckptName := ctx.Query("ckpt_name")
@@ -838,7 +839,7 @@ func createForGPU(ctx *context.Context, jobName string) error {
 	evaluationIndex := ctx.Query("evaluation_index")
 	Params := ctx.Query("run_para_list")
 	specId := ctx.QueryInt64("spec_id")
-	TrainUrl := ctx.Query("train_url")
+	TrainUrl := ctx.Query("pre_train_model_url")
 	CkptName := ctx.Query("ckpt_name")
 	modelName := ctx.Query("model_name")
 	modelVersion := ctx.Query("model_version")
@@ -968,10 +969,9 @@ func modelSafetyNewDataPrepare(ctx *context.Context) error {
 	ctx.Data["evaluationIndex"] = ctx.Query("evaluation_index")
 	ctx.Data["run_para_list"] = ctx.Query("run_para_list")
 	ctx.Data["spec_id"] = ctx.QueryInt64("spec_id")
-	ctx.Data["train_url"] = ctx.Query("train_url")
+	ctx.Data["pre_train_model_url"] = ctx.Query("pre_train_model_url")
 	ctx.Data["ckpt_name"] = ctx.Query("ckpt_name")
-
-	ctx.Data["train_url"] = ctx.Query("train_url")
+	ctx.Data["pre_train_model_url"] = ctx.Query("pre_train_model_url")
 	ctx.Data["ckpt_name"] = ctx.Query("ckpt_name")
 	ctx.Data["model_name"] = ctx.Query("model_name")
 	ctx.Data["model_version"] = ctx.Query("model_version")