diff --git a/services/ai_task_service/cluster/cloudbrain_two.go b/services/ai_task_service/cluster/cloudbrain_two.go
index 162753b017..58ca694293 100644
--- a/services/ai_task_service/cluster/cloudbrain_two.go
+++ b/services/ai_task_service/cluster/cloudbrain_two.go
@@ -2,6 +2,14 @@ package cluster
 
 import "C"
 import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"path"
+	"strconv"
+	"strings"
+
 	"code.gitea.io/gitea/entity"
 	"code.gitea.io/gitea/manager/client/cloudbrain_two"
 	"code.gitea.io/gitea/manager/client/cloudbrain_two_cd"
@@ -11,13 +19,6 @@ import (
 	"code.gitea.io/gitea/modules/setting"
 	"code.gitea.io/gitea/modules/timeutil"
 	"code.gitea.io/gitea/services/ai_task_service/storage_helper"
-	"encoding/json"
-	"errors"
-	"fmt"
-	"io"
-	"path"
-	"strconv"
-	"strings"
 )
 
 type CloudbrainTwoClusterAdapter struct {
@@ -475,7 +476,7 @@ func handleCloudbrainTwoParameter(req entity.CreateTrainTaskRequest) models.Para
 			Value: multiModelUrl,
 		}, models.Parameter{
 			Label: modelarts.CkptUrl,
-			Value: t.Datasets[0].S3DownloadUrl,
+			Value: t.PreTrainModel[0].S3DownloadUrl,
 		})
 	}