diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go
index 3ca6cf1157..feacb45472 100644
--- a/routers/repo/ai_model_manage.go
+++ b/routers/repo/ai_model_manage.go
@@ -97,7 +97,7 @@ func saveModelByParameters(jobId string, versionName string, name string, versio
 		UserId:        ctx.User.ID,
 		CodeBranch:    aiTask.BranchName,
 		CodeCommitID:  aiTask.CommitID,
-		Engine:        aiTask.EngineID,
+		Engine:        int64(engine),
 		TrainTaskInfo: string(aiTaskJson),
 		Accuracy:      string(accuracyJson),
 	}