feat: update stable diffusion api version

This commit is contained in:
RockYang 2023-11-08 14:54:10 +08:00
parent 696306f066
commit e221b1eed4
4 changed files with 34 additions and 62 deletions

View File

@ -1,37 +1,38 @@
{ {
"data": [ "data": [
"task(38194gitxp745ha)", "task(m1wpaa4v60zedj8)",
"A beautiful Chinese girl riding on a tiger", "a cute cat",
"", "",
[], [],
20, 20,
"Euler a", "DPM++ 2M Karras",
false,
false,
1, 1,
1, 1,
7, 7,
-1,
-1,
0,
0,
0,
false,
512,
512, 512,
384,
true, true,
0.7, 0.7,
2, 2,
"ESRGAN_4x", "ESRGAN_4x",
10,
0, 0,
0, 0,
0, "Use same checkpoint",
"Use same sampler", "Use same sampler",
"", "",
"", "",
[], [],
"None", "None",
null, false,
"",
0.8,
-1,
false,
-1,
0,
0,
0,
false, false,
false, false,
"positive", "positive",
@ -54,45 +55,13 @@
false, false,
false, false,
0, 0,
"Not set",
true,
true,
"",
"",
"",
"",
"",
1.3,
"Not set",
"Not set",
1.3,
"Not set",
1.3,
"Not set",
1.3,
1.3,
"Not set",
1.3,
"Not set",
1.3,
"Not set",
1.3,
"Not set",
1.3,
"Not set",
1.3,
"Not set",
false, false,
"None",
null,
false,
50,
[], [],
"", "",
"", "",
"" ""
], ],
"event_data": null, "event_data": null,
"fn_index": 232, "fn_index": 96,
"session_hash": "3xedmn4nuzq" "session_hash": "kmb0ojjfhdj"
} }

View File

@ -105,7 +105,8 @@ func (s *Service) Txt2Img(task types.SdTask) error {
data[ParamKeys["negative_prompt"]] = params.NegativePrompt data[ParamKeys["negative_prompt"]] = params.NegativePrompt
data[ParamKeys["steps"]] = params.Steps data[ParamKeys["steps"]] = params.Steps
data[ParamKeys["sampler"]] = params.Sampler data[ParamKeys["sampler"]] = params.Sampler
data[ParamKeys["face_fix"]] = params.FaceFix // @fix bug: 有些 stable diffusion 没有面部修复功能
//data[ParamKeys["face_fix"]] = params.FaceFix
data[ParamKeys["cfg_scale"]] = params.CfgScale data[ParamKeys["cfg_scale"]] = params.CfgScale
data[ParamKeys["seed"]] = params.Seed data[ParamKeys["seed"]] = params.Seed
data[ParamKeys["height"]] = params.Height data[ParamKeys["height"]] = params.Height
@ -176,7 +177,8 @@ func (s *Service) runTask(taskInfo TaskInfo, client *req.Client) {
var info map[string]any var info map[string]any
err = utils.JsonDecode(utils.InterfaceToString(res.Data[1]), &info) err = utils.JsonDecode(utils.InterfaceToString(res.Data[1]), &info)
if err != nil { if err != nil {
cbReq.Message = err.Error() logger.Error(res.Data)
cbReq.Message = "error with decode image url:" + err.Error()
cbReq.Success = false cbReq.Success = false
result <- cbReq result <- cbReq
return return
@ -229,6 +231,7 @@ 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)
} }

View File

@ -32,14 +32,14 @@ var ParamKeys = map[string]int{
"negative_prompt": 2, "negative_prompt": 2,
"steps": 4, "steps": 4,
"sampler": 5, "sampler": 5,
"face_fix": 6, "face_fix": 6, // 面部修复
"cfg_scale": 10, "cfg_scale": 8,
"seed": 11, "seed": 27,
"height": 17, "height": 9,
"width": 18, "width": 10,
"hd_fix": 19, "hd_fix": 11,
"hd_redraw_rate": 20, //高清修复重绘幅度 "hd_redraw_rate": 12, //高清修复重绘幅度
"hd_scale": 21, // 高清修复放大倍数 "hd_scale": 13, // 高清修复放大倍数
"hd_scale_alg": 22, // 高清修复放大算法 "hd_scale_alg": 14, // 高清修复放大算法
"hd_sample_num": 23, // 高清修复采样次数 "hd_sample_num": 15, // 高清修复采样次数
} }

View File

@ -148,11 +148,11 @@ func IntValue(str string, defaultValue int) int {
} }
func ForceCovert(src any, dst interface{}) error { func ForceCovert(src any, dst interface{}) error {
bytes, err := json.Marshal(src) b, err := json.Marshal(src)
if err != nil { if err != nil {
return err return err
} }
err = json.Unmarshal(bytes, dst) err = json.Unmarshal(b, dst)
if err != nil { if err != nil {
return err return err
} }