diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go
index 2d001ada0..9efbb11c4 100644
--- a/routers/repo/ai_model_manage.go
+++ b/routers/repo/ai_model_manage.go
@@ -104,7 +104,7 @@ func saveModelByParameters(jobId string, versionName string, name string, versio
 	var units []models.RepoUnit
 	var deleteUnitTypes []models.UnitType
 	units = append(units, models.RepoUnit{
-		RepoID: ctx.Repo.Repository.RepoID,
+		RepoID: ctx.Repo.Repository.ID,
 		Type:   models.UnitTypeModelManage,
 		Config: &models.ModelManageConfig{
 			EnableModelManage: true,