mirror of
				https://github.com/yangjian102621/geekai.git
				synced 2025-11-04 16:23:42 +08:00 
			
		
		
		
	feat: update stable diffusion api version
This commit is contained in:
		@@ -1,37 +1,38 @@
 | 
			
		||||
{
 | 
			
		||||
  "data": [
 | 
			
		||||
    "task(38194gitxp745ha)",
 | 
			
		||||
    "A beautiful Chinese girl riding on a tiger",
 | 
			
		||||
    "task(m1wpaa4v60zedj8)",
 | 
			
		||||
    "a cute cat",
 | 
			
		||||
    "",
 | 
			
		||||
    [],
 | 
			
		||||
    20,
 | 
			
		||||
    "Euler a",
 | 
			
		||||
    false,
 | 
			
		||||
    false,
 | 
			
		||||
    "DPM++ 2M Karras",
 | 
			
		||||
    1,
 | 
			
		||||
    1,
 | 
			
		||||
    7,
 | 
			
		||||
    -1,
 | 
			
		||||
    -1,
 | 
			
		||||
    0,
 | 
			
		||||
    0,
 | 
			
		||||
    0,
 | 
			
		||||
    false,
 | 
			
		||||
    512,
 | 
			
		||||
    512,
 | 
			
		||||
    384,
 | 
			
		||||
    true,
 | 
			
		||||
    0.7,
 | 
			
		||||
    2,
 | 
			
		||||
    "ESRGAN_4x",
 | 
			
		||||
    10,
 | 
			
		||||
    0,
 | 
			
		||||
    0,
 | 
			
		||||
    0,
 | 
			
		||||
    "Use same checkpoint",
 | 
			
		||||
    "Use same sampler",
 | 
			
		||||
    "",
 | 
			
		||||
    "",
 | 
			
		||||
    [],
 | 
			
		||||
    "None",
 | 
			
		||||
    null,
 | 
			
		||||
    false,
 | 
			
		||||
    "",
 | 
			
		||||
    0.8,
 | 
			
		||||
    -1,
 | 
			
		||||
    false,
 | 
			
		||||
    -1,
 | 
			
		||||
    0,
 | 
			
		||||
    0,
 | 
			
		||||
    0,
 | 
			
		||||
    false,
 | 
			
		||||
    false,
 | 
			
		||||
    "positive",
 | 
			
		||||
@@ -54,45 +55,13 @@
 | 
			
		||||
    false,
 | 
			
		||||
    false,
 | 
			
		||||
    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,
 | 
			
		||||
    "None",
 | 
			
		||||
    null,
 | 
			
		||||
    false,
 | 
			
		||||
    50,
 | 
			
		||||
    [],
 | 
			
		||||
    "",
 | 
			
		||||
    "",
 | 
			
		||||
    ""
 | 
			
		||||
  ],
 | 
			
		||||
  "event_data": null,
 | 
			
		||||
  "fn_index": 232,
 | 
			
		||||
  "session_hash": "3xedmn4nuzq"
 | 
			
		||||
  "fn_index": 96,
 | 
			
		||||
  "session_hash": "kmb0ojjfhdj"
 | 
			
		||||
}
 | 
			
		||||
@@ -105,7 +105,8 @@ func (s *Service) Txt2Img(task types.SdTask) error {
 | 
			
		||||
	data[ParamKeys["negative_prompt"]] = params.NegativePrompt
 | 
			
		||||
	data[ParamKeys["steps"]] = params.Steps
 | 
			
		||||
	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["seed"]] = params.Seed
 | 
			
		||||
	data[ParamKeys["height"]] = params.Height
 | 
			
		||||
@@ -176,7 +177,8 @@ func (s *Service) runTask(taskInfo TaskInfo, client *req.Client) {
 | 
			
		||||
		var info map[string]any
 | 
			
		||||
		err = utils.JsonDecode(utils.InterfaceToString(res.Data[1]), &info)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			cbReq.Message = err.Error()
 | 
			
		||||
			logger.Error(res.Data)
 | 
			
		||||
			cbReq.Message = "error with decode image url:" + err.Error()
 | 
			
		||||
			cbReq.Success = false
 | 
			
		||||
			result <- cbReq
 | 
			
		||||
			return
 | 
			
		||||
@@ -229,6 +231,7 @@ func (s *Service) runTask(taskInfo TaskInfo, client *req.Client) {
 | 
			
		||||
 | 
			
		||||
			cbReq.ImageData = progressRes.LivePreview
 | 
			
		||||
			cbReq.Progress = int(progressRes.Progress * 100)
 | 
			
		||||
			logger.Debug(cbReq)
 | 
			
		||||
			s.callback(cbReq)
 | 
			
		||||
			time.Sleep(time.Second)
 | 
			
		||||
		}
 | 
			
		||||
 
 | 
			
		||||
@@ -32,14 +32,14 @@ var ParamKeys = map[string]int{
 | 
			
		||||
	"negative_prompt": 2,
 | 
			
		||||
	"steps":           4,
 | 
			
		||||
	"sampler":         5,
 | 
			
		||||
	"face_fix":        6,
 | 
			
		||||
	"cfg_scale":       10,
 | 
			
		||||
	"seed":            11,
 | 
			
		||||
	"height":          17,
 | 
			
		||||
	"width":           18,
 | 
			
		||||
	"hd_fix":          19,
 | 
			
		||||
	"hd_redraw_rate":  20, //高清修复重绘幅度
 | 
			
		||||
	"hd_scale":        21, // 高清修复放大倍数
 | 
			
		||||
	"hd_scale_alg":    22, // 高清修复放大算法
 | 
			
		||||
	"hd_sample_num":   23, // 高清修复采样次数
 | 
			
		||||
	"face_fix":        6, // 面部修复
 | 
			
		||||
	"cfg_scale":       8,
 | 
			
		||||
	"seed":            27,
 | 
			
		||||
	"height":          9,
 | 
			
		||||
	"width":           10,
 | 
			
		||||
	"hd_fix":          11,
 | 
			
		||||
	"hd_redraw_rate":  12, //高清修复重绘幅度
 | 
			
		||||
	"hd_scale":        13, // 高清修复放大倍数
 | 
			
		||||
	"hd_scale_alg":    14, // 高清修复放大算法
 | 
			
		||||
	"hd_sample_num":   15, // 高清修复采样次数
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -148,11 +148,11 @@ func IntValue(str string, defaultValue int) int {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func ForceCovert(src any, dst interface{}) error {
 | 
			
		||||
	bytes, err := json.Marshal(src)
 | 
			
		||||
	b, err := json.Marshal(src)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	err = json.Unmarshal(bytes, dst)
 | 
			
		||||
	err = json.Unmarshal(b, dst)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user