diff --git a/models/Model.go b/models/Model.go index a6da634..7734c46 100644 --- a/models/Model.go +++ b/models/Model.go @@ -85,7 +85,20 @@ func (model *Model) Inference(image_list []Image, callback func(Image)) { // 执行生成任务 if model.Image == "" { + img := image_list[0] + // 发送的参数 + var datx map[string]interface{} = make(map[string]interface{}) + datx["prompt"] = img.Prompt // 提示词 + datx["seed"] = img.Seed // 随机数种子 + datx["n_iter"] = len(image_list) // 生成图像数量 + datx["steps"] = 50 // 迭代步数 + datx["cfg_scale"] = img.CfgScale // 提示词引导系数 (CFG Scale) + if img.SamplerName == "" { + datx["sampler_name"] = img.SamplerName // 采样器名称 + } + fmt.Println("image_list:", datx) + var data = struct { //EnableHr bool `json:"enable_hr"` //DenoisingStrength int `json:"denoising_strength"` @@ -179,12 +192,14 @@ func (model *Model) Inference(image_list []Image, callback func(Image)) { //SaveImages: false, //AlwaysonScripts: map[string]string{}, } + fmt.Println("data:", data) + // 接收到的图片列表 var rest = struct { Images []string `json:"images"` }{} var url = fmt.Sprintf("http://%s:%d/sdapi/v1/txt2img", server.IP, server.Port) - if err := goreq.Post(url).SetJsonBody(data).Do().BindJSON(&rest); err != nil { + if err := goreq.Post(url).SetJsonBody(datx).Do().BindJSON(&rest); err != nil { log.Println("API 查询失败:", err) } for index, img := range rest.Images { diff --git a/routers/images.go b/routers/images.go index 6164416..6cbdaf5 100644 --- a/routers/images.go +++ b/routers/images.go @@ -137,9 +137,6 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) { if template.CfgScale > 20 { template.CfgScale = 20 } - if template.SamplerName == "" { - template.SamplerName = "DDIM" - } if template.ModelID <= 0 { w.WriteHeader(http.StatusBadRequest) w.Write([]byte("model_id 参数不能为空"))