diff --git a/models/ai_model_manage.go b/models/ai_model_manage.go
index c1ab3afab..502d3aadb 100644
--- a/models/ai_model_manage.go
+++ b/models/ai_model_manage.go
@@ -242,30 +242,16 @@ func QueryModelConvertById(id string) (*AiModelConvert, error) {
 	if isExist {
 		return re, nil
 	}
-	//sess.Select("*").Table(new(AiModelConvert)).Where("id = ?", id)
-	//aiModelManageConvertList := make([]*AiModelConvert, 0)
-	//err := sess.Find(&aiModelManageConvertList)
-	//if err == nil {
-	//		if len(aiModelManageConvertList) == 1 {
-	//		return aiModelManageConvertList[0], nil
-	//	}
-	//}
 	return nil, err
 }
 
 func QueryModelById(id string) (*AiModelManage, error) {
 	sess := x.NewSession()
 	defer sess.Close()
-	sess.Select("*").Table("ai_model_manage").
-		Where("id='" + id + "'")
-	aiModelManageList := make([]*AiModelManage, 0)
-	err := sess.Find(&aiModelManageList)
-	if err == nil {
-		if len(aiModelManageList) == 1 {
-			return aiModelManageList[0], nil
-		}
-	} else {
-		log.Info("error=" + err.Error())
+	re := new(AiModelManage)
+	isExist, err := sess.Table(new(AiModelManage)).ID(id).Get(re)
+	if isExist {
+		return re, nil
 	}
 	return nil, err
 }