xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-12-27 18:45:59 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor: use waterflow component in mj, sd and dall image drawing page

Browse Source
This commit is contained in:
RockYang
2024-05-13 19:04:00 +08:00
parent 5fb9f84182
commit 43b6665370
19 changed files with 1118 additions and 759 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/service/sd/service.go
View File

@@ -60,7 +60,7 @@ func (s *Service) Run() {
logger.Errorf("taking task with error: %v", err)
continue
}
logger.Infof("%s handle a new Stable-Diffusion task: %+v", s.name, task)
// translate prompt
if utils.HasChinese(task.Params.Prompt) {
content, err := utils.OpenAIRequest(s.db, fmt.Sprintf(service.RewritePromptTemplate, task.Params.Prompt))