采样器

routers/images.go

@@ -106,9 +106,9 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) {
 				FromImage      int    `json:"from_image"`      // 来源图片(图生图时使用)
 				Prompt         string `json:"prompt"`          // 提示词
 				NegativePrompt string `json:"negative_prompt"` // 负面提示词
-				Steps          int    `json:"steps"`           // 推理步数
-				CfgScale       int    `json:"cfg_scale"`       // 引导比例
-				Scheduler      string `json:"scheduler"`       // 调度器
+				Steps          int    `json:"steps"`           // 迭代步数
+				CfgScale       int    `json:"cfg_scale"`       // 提示词引导系数 (CFG Scale)
+				SamplerName    string `json:"sampler_name"`    // 采样器名称(Sampler Name)
 				Seed           int    `json:"seed"`            // 随机种子(单张图生成时使用)
 				NIter          int    `json:"n_iter"`          // 生成数量
 				ModelID        int    `json:"model_id"`        // 模型ID
@@ -137,8 +137,8 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) {
 			if template.CfgScale > 20 {
 				template.CfgScale = 20
 			}
-			if template.Scheduler == "" {
-				template.Scheduler = "DDIM"
+			if template.SamplerName == "" {
+				template.SamplerName = "DDIM"
 			}
 			if template.ModelID <= 0 {
 				w.WriteHeader(http.StatusBadRequest)
@@ -167,7 +167,7 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) {
 				image.NegativePrompt = template.NegativePrompt
 				image.NumInferenceSteps = template.Steps
 				image.CfgScale = template.CfgScale
-				image.Scheduler = template.Scheduler
+				image.SamplerName = template.SamplerName
 				image.Seed = template.Seed
 				image_list = append(image_list, image)
 			}