采样器

models/Image.go

@@ -29,7 +29,7 @@ type Image struct {
 	NegativePrompt    string    `json:"negative_prompt"`       // 负向提示
 	NumInferenceSteps int       `json:"num_inference_steps"`   // 推理步数(minimum: 1; maximum: 500)
 	CfgScale          int       `json:"cfg_scale"`             // 引导比例(minimum: 1; maximum: 20)
-	Scheduler         string    `json:"scheduler"`             // 调度器(DDIM|K_EULER|DPMSolverMultistep|K_EULER_ANCESTRAL|PNDM|KLMS)
+	SamplerName       string    `json:"sampler_name"`          // 采样器名称
 	Seed              int       `json:"seed"`                  // 随机种子(minimum: 0; maximum: 2147483647)
 	FromImage         int       `json:"from_image"`            // 来源图片(如果是从图片生成的, 则记录来源图片的ID)
 	Task              string    `json:"task"`                  // 任务编号(uuid)

models/Model.go

@@ -106,7 +106,7 @@ func (model *Model) Inference(image_list []Image, callback func(Image)) {
 				//SubseedStrength                   int               `json:"subseed_strength"`
 				//SeedResizeFromH                   int               `json:"seed_resize_from_h"`
 				//SeedResizeFromW                   int               `json:"seed_resize_from_w"`
-				//SamplerName                       string            `json:"sampler_name"`
+				SamplerName string `json:"sampler_name"`
 				//BatchSize                         int               `json:"batch_size"`
 				NIter    int `json:"n_iter"`
 				Steps    int `json:"steps"`
@@ -152,7 +152,7 @@ func (model *Model) Inference(image_list []Image, callback func(Image)) {
 				//SubseedStrength:                   0,
 				//SeedResizeFromH:                   -1,
 				//SeedResizeFromW:                   -1,
-				//SamplerName:                       "beamsearch",
+				SamplerName: image_list[0].SamplerName, // 采样器名称
 				//BatchSize:                         1,
 				NIter:    len(image_list), // 1~100
 				Steps:    50,              // 1~150