diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go index 362f6e7c14..2767f1594c 100644 --- a/routers/repo/aisafety.go +++ b/routers/repo/aisafety.go @@ -823,6 +823,10 @@ func createForNPU(ctx *context.Context, jobName string) error { JobType: string(models.JobTypeModelSafety), } + userCommand, userImageUrl := getInfJobUserCommand(engineID, req) + req.UserCommand = userCommand + req.UserImageUrl = userImageUrl + _, err = modelarts.GenerateInferenceJob(ctx, req) if err != nil { log.Error("GenerateTrainJob failed:%v", err.Error())