xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-09-18 01:06:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: change default params for stable diffusion

Browse Source
This commit is contained in:
RockYang 2023-12-10 14:45:22 +08:00
parent 4a9f7e3bce
commit 024c0032eb
3 changed files with 24 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
api/res/text2img.json
View File

@ -1,20 +1,20 @@
{ {
"data": [ "data": [
"task(m1wpaa4v60zedj8)", "task(s95jqt5jr8yppcp)",
"a cute cat", "A beautiful Chinese girl in a garden",
"", "",
[], [],
20, 30,
"DPM++ 2M Karras", "Euler a",
1, 1,
1, 1,
7, 7,
512, 512,
384, 512,
true, true,
0.7, 0.7,
2, 2,
"ESRGAN_4x", "Latent",
10, 10,
0, 0,
0, 0,
@ -55,13 +55,13 @@
false, false,
false, false,
0, 0,
false, [
[], ],
"", "",
"", "",
"" ""
], ],
"event_data": null, "event_data": null,
"fn_index": 96, "fn_index": 95,
"session_hash": "kmb0ojjfhdj" "session_hash": "eqwumnt3rov"
} }

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

@ -135,7 +135,6 @@ func (s *Service) runTask(taskInfo TaskInfo, client *req.Client) {
"fn_index": taskInfo.FnIndex, "fn_index": taskInfo.FnIndex,
"session_hash": taskInfo.SessionHash, "session_hash": taskInfo.SessionHash,
} }
logger.Debug(utils.JsonEncode(body))
var result = make(chan CBReq) var result = make(chan CBReq)
go func() { go func() {
var res struct { var res struct {
@ -231,7 +230,6 @@ func (s *Service) runTask(taskInfo TaskInfo, client *req.Client) {
cbReq.ImageData = progressRes.LivePreview cbReq.ImageData = progressRes.LivePreview
cbReq.Progress = int(progressRes.Progress * 100) cbReq.Progress = int(progressRes.Progress * 100)
logger.Debug(cbReq)
s.callback(cbReq) s.callback(cbReq)
time.Sleep(time.Second) time.Sleep(time.Second)
} }
@ -287,8 +285,13 @@ func (s *Service) callback(data CBReq) {
if data.Progress < 100 && data.ImageData != "" { if data.Progress < 100 && data.ImageData != "" {
jobVo.ImgURL = data.ImageData jobVo.ImgURL = data.ImageData
} }
logger.Infof("绘图进度：%d", data.Progress)
// 扣减绘图次数 // 扣减绘图次数
s.db.Model(&model.User{}).Where("id = ?", jobVo.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1)) if data.Progress == 100 {
s.db.Model(&model.User{}).Where("id = ? AND img_calls > 0", jobVo.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
}
// 推送任务到前端 // 推送任务到前端
if client != nil { if client != nil {
utils.ReplyChunkMessage(client, jobVo) utils.ReplyChunkMessage(client, jobVo)

16
web/src/views/ImageSd.vue
View File

@ -500,21 +500,21 @@ window.onresize = () => {
} }
const samplers = ["Euler a", "Euler", "DPM2 a Karras", "DPM++ 2S a Karras", "DPM++ 2M Karras", "DPM++ SDE Karras", "DPM2", "DPM2 a", "DPM++ 2S a", "DPM++ 2M", "DPM++ SDE", "DPM fast", "DPM adaptive", const samplers = ["Euler a", "Euler", "DPM2 a Karras", "DPM++ 2S a Karras", "DPM++ 2M Karras", "DPM++ SDE Karras", "DPM2", "DPM2 a", "DPM++ 2S a", "DPM++ 2M", "DPM++ SDE", "DPM fast", "DPM adaptive",
"LMS Karras", "DPM2 Karras", "DDIM", "PLMS", "UniPC", "LMS", "Heun",] "LMS Karras", "DPM2 Karras", "DDIM", "PLMS", "UniPC", "LMS", "Heun",]
const scaleAlg = ["ESRGAN_4x", "R-ESRGAN 4x+", "SwinIR_4x", "LDSR"] const scaleAlg = ["Latent", "ESRGAN_4x", "R-ESRGAN 4x+", "SwinIR_4x", "LDSR"]
const params = ref({ const params = ref({
width: 1024, width: 512,
height: 1024, height: 512,
sampler: samplers[0], sampler: samplers[0],
seed: -1, seed: -1,
steps: 20, steps: 30,
cfg_scale: 7, cfg_scale: 7,
face_fix: false, face_fix: false,
hd_fix: false, hd_fix: true,
hd_redraw_rate: 0.3, hd_redraw_rate: 0.7,
hd_scale: 2, hd_scale: 2,
hd_scale_alg: scaleAlg[0], hd_scale_alg: scaleAlg[0],
hd_steps: 0, hd_steps: 10,
prompt: "A beautiful Chinese girl riding on a tiger", prompt: "A beautiful Chinese girl in a garden",
negative_prompt: "nsfw, paintings, cartoon, anime, sketches, low quality,easynegative,ng_deepnegative _v1 75t,(worst quality:2),(low quality:2),(normalquality:2),lowres,bad anatomy,bad hands,normal quality,((monochrome)),((grayscale)),((watermark))", negative_prompt: "nsfw, paintings, cartoon, anime, sketches, low quality,easynegative,ng_deepnegative _v1 75t,(worst quality:2),(low quality:2),(normalquality:2),lowres,bad anatomy,bad hands,normal quality,((monochrome)),((grayscale)),((watermark))",
}) })