diff --git a/api/service/dalle/service.go b/api/service/dalle/service.go
index 2306f446..ba716bdc 100644
--- a/api/service/dalle/service.go
+++ b/api/service/dalle/service.go
@@ -132,7 +132,11 @@ func (s *Service) Image(task types.DallTask, sync bool) (string, error) {
 	}
 
 	var chatModel model.ChatModel
-	s.db.Where("id = ?", task.ModelId).First(&chatModel)
+	if task.ModelId > 0 {
+		s.db.Where("id", task.ModelId).First(&chatModel)
+	} else {
+		s.db.Where("value", task.ModelName).First(&chatModel)
+	}
 
 	// get image generation API KEY
 	var apiKey model.ApiKey