迭代步数

models/Image.go

@@ -27,7 +27,7 @@ type Image struct {
 	Format         string    `json:"format"`                // 格式
 	Prompt         string    `json:"prompt"`                // 提示词
 	NegativePrompt string    `json:"negative_prompt"`       // 负向提示
-	NumInferenceSteps int       `json:"num_inference_steps"`   // 推理步数(minimum: 1; maximum: 500)
+	Steps          int       `json:"steps"`                 // 迭代步数 (Steps 1~150)
 	CfgScale       int       `json:"cfg_scale"`             // 引导比例(minimum: 1; maximum: 20)
 	SamplerName    string    `json:"sampler_name"`          // 采样器名称
 	Seed           int       `json:"seed"`                  // 随机种子(minimum: 0; maximum: 2147483647)

models/Model.go

@@ -92,7 +92,7 @@ func (model *Model) Inference(image_list []Image, callback func(Image)) {
 			datx["prompt"] = img.Prompt      // 提示词
 			datx["seed"] = img.Seed          // 随机数种子
 			datx["n_iter"] = len(image_list) // 生成图像数量
-			datx["steps"] = 50               // 迭代步数
+			datx["steps"] = img.Steps        // 迭代步数
 			datx["cfg_scale"] = img.CfgScale // 提示词引导系数 (CFG Scale)
 			if img.SamplerName == "" {
 				datx["sampler_name"] = img.SamplerName // 采样器名称

routers/images.go

@@ -162,7 +162,7 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) {
 				image.FromImage = template.FromImage
 				image.Prompt = template.Prompt
 				image.NegativePrompt = template.NegativePrompt
-				image.NumInferenceSteps = template.Steps
+				image.Steps = template.Steps
 				image.CfgScale = template.CfgScale
 				image.SamplerName = template.SamplerName
 				image.Seed = template.Seed