mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-18 17:26:38 +08:00
feat: update stable diffusion api version
This commit is contained in:
parent
696306f066
commit
e221b1eed4
@ -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"
|
||||||
}
|
}
|
@ -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)
|
||||||
}
|
}
|
||||||
|
@ -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, // 高清修复采样次数
|
||||||
}
|
}
|
||||||
|
@ -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
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user