diff --git a/routers/repo/ai_model_convert.go b/routers/repo/ai_model_convert.go index ddf04b2c4..78e8c6ac5 100644 --- a/routers/repo/ai_model_convert.go +++ b/routers/repo/ai_model_convert.go @@ -76,10 +76,16 @@ func SaveModelConvert(ctx *context.Context) { UserId: ctx.User.ID, } models.SaveModelConvert(modelConvert) - createTrainJob(modelConvert, ctx, task.Path) - ctx.JSON(200, map[string]string{ - "result_code": "0", - }) + err = createTrainJob(modelConvert, ctx, task.Path) + if err == nil { + ctx.JSON(200, map[string]string{ + "result_code": "0", + }) + } else { + ctx.JSON(200, map[string]string{ + "result_code": "1", + }) + } } func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, modelRelativePath string) error {