diff --git a/services/ai_task_service/task/task_service.go b/services/ai_task_service/task/task_service.go
index 12dbc3587f..a0910343ac 100644
--- a/services/ai_task_service/task/task_service.go
+++ b/services/ai_task_service/task/task_service.go
@@ -761,7 +761,7 @@ func GetModelDownload(task *models.Cloudbrain) []*models.ModelDownload {
 	} else {
 		model, err = models.QueryModelById(task.ModelId)
 	}
-	if err != nil {
+	if err != nil || model == nil {
 		return pretrainModelList
 	}