提示词引导系数 (CFG Scale)

models/Image.go

@@ -28,7 +28,7 @@ type Image struct {
 	Prompt            string    `json:"prompt"`                // 提示词
 	NegativePrompt    string    `json:"negative_prompt"`       // 负向提示
 	NumInferenceSteps int       `json:"num_inference_steps"`   // 推理步数(minimum: 1; maximum: 500)
-	GuidanceScale     float32   `json:"guidance_scale"`        // 引导比例(minimum: 1; maximum: 20)
+	CfgScale          int       `json:"cfg_scale"`             // 引导比例(minimum: 1; maximum: 20)
 	Scheduler         string    `json:"scheduler"`             // 调度器(DDIM|K_EULER|DPMSolverMultistep|K_EULER_ANCESTRAL|PNDM|KLMS)
 	Seed              string    `json:"seed"`                  // 随机种子(minimum: 0; maximum: 2147483647)
 	FromImage         int       `json:"from_image"`            // 来源图片(如果是从图片生成的, 则记录来源图片的ID)

models/Model.go

@@ -110,7 +110,7 @@ func (model *Model) Inference(image_list []Image, callback func(Image)) {
 				//BatchSize                         int               `json:"batch_size"`
 				NIter    int `json:"n_iter"`
 				Steps    int `json:"steps"`
-				//CfgScale                          int               `json:"cfg_scale"`
+				CfgScale int `json:"cfg_scale"`
 				//Width                             int               `json:"width"`
 				//Height                            int               `json:"height"`
 				//RestoreFaces                      bool              `json:"restore_faces"`
@@ -154,9 +154,9 @@ func (model *Model) Inference(image_list []Image, callback func(Image)) {
 				//SeedResizeFromW:                   -1,
 				//SamplerName:                       "beamsearch",
 				//BatchSize:                         1,
-				NIter: len(image_list),
-				Steps: 50,
-				//CfgScale:                          7,
+				NIter:    len(image_list), // 1~100
+				Steps:    50,              // 1~150
+				CfgScale: image_list[0].CfgScale,
 				//Width:                             512,
 				//Height:                            512,
 				//RestoreFaces:                      false,

routers/images.go

@@ -107,7 +107,7 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) {
 				Prompt         string `json:"prompt"`          // 提示词
 				NegativePrompt string `json:"negative_prompt"` // 负面提示词
 				Steps          int    `json:"steps"`           // 推理步数
-				GuidanceScale  float32 `json:"guidance_scale"`  // 引导比例
+				CfgScale       int    `json:"cfg_scale"`       // 引导比例
 				Scheduler      string `json:"scheduler"`       // 调度器
 				Seed           string `json:"seed"`            // 随机种子(单张图生成时使用)
 				NIter          int    `json:"n_iter"`          // 生成数量
@@ -131,11 +131,11 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) {
 			if template.Steps <= 0 {
 				template.Steps = 50
 			}
-			if template.GuidanceScale <= 0 {
-				template.GuidanceScale = 1
+			if template.CfgScale <= 0 {
+				template.CfgScale = 1
 			}
-			if template.GuidanceScale > 20 {
-				template.GuidanceScale = 20
+			if template.CfgScale > 20 {
+				template.CfgScale = 20
 			}
 			if template.Scheduler == "" {
 				template.Scheduler = "DDIM"
@@ -166,7 +166,7 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) {
 				image.Prompt = template.Prompt
 				image.NegativePrompt = template.NegativePrompt
 				image.NumInferenceSteps = template.Steps
-				image.GuidanceScale = template.GuidanceScale
+				image.CfgScale = template.CfgScale
 				image.Scheduler = template.Scheduler
 				image.Seed = template.Seed
 				image_list = append(image_list, image)