diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go index c69238eea..076de48db 100644 --- a/routers/repo/aisafety.go +++ b/routers/repo/aisafety.go @@ -4,6 +4,7 @@ import ( "bufio" "encoding/json" "errors" + "fmt" "io" "io/ioutil" "net/http" @@ -841,19 +842,20 @@ func createForNPU(ctx *context.Context, jobName string) { Params := ctx.Query("run_para_list") specId := ctx.QueryInt64("spec_id") - engineID := ctx.QueryInt("EngineID") - poolID := ctx.Query("PoolID") + engineID := ctx.QueryInt("engine_id") + log.Info("engine_id=" + fmt.Sprint(engineID)) + poolID := ctx.Query("pool_id") repo := ctx.Repo.Repository trainUrl := ctx.Query("train_url") - modelName := ctx.Query("ModelName") - modelVersion := ctx.Query("ModelVersion") + modelName := ctx.Query("model_name") + modelVersion := ctx.Query("model_version") ckptName := ctx.Query("ckpt_name") ckptUrl := "/" + trainUrl + ckptName log.Info("ckpt url:" + ckptUrl) - FlavorName := ctx.Query("FlavorName") - EngineName := ctx.Query("EngineName") + FlavorName := ctx.Query("flaver_names") + EngineName := ctx.Query("engine_names") isLatestVersion := modelarts.IsLatestVersion VersionCount := modelarts.VersionCountOne