#5213 fix模型安全评测

Merged
liuzx merged 1 commits from path_unite_all into V20240116 2 months ago
  1. +4
    -0
      routers/repo/aisafety.go

+ 4
- 0
routers/repo/aisafety.go View File

@@ -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())


Loading…
Cancel
Save