精简表单

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 {