diff --git a/routers/repo/ai_model_convert.go b/routers/repo/ai_model_convert.go index fcc8aecb9..ddf04b2c4 100644 --- a/routers/repo/ai_model_convert.go +++ b/routers/repo/ai_model_convert.go @@ -83,7 +83,7 @@ func SaveModelConvert(ctx *context.Context) { } func createTrainJob(modelConvert *models.AiModelConvert, ctx *context.Context, modelRelativePath string) error { - repo, _ := models.GetRepositoryByID(ctx.Repo.Repository.RepoID) + repo, _ := models.GetRepositoryByID(ctx.Repo.Repository.ID) if modelConvert.SrcEngine == PYTORCH_ENGINE { codePath := setting.JobPath + modelConvert.ID + CodeMountPath downloadCode(repo, codePath, DefaultBranchName)