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