gameui/ai
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

迭代步数

Browse Source
This commit is contained in:
桜華
2023-06-23 17:59:45 +08:00
parent ac45eafc87
commit 4cf1b91c2c
3 changed files with 25 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
models/Image.go
View File

@@ -27,7 +27,7 @@ type Image struct {
Format string `json:"format"` // 格式 Format string `json:"format"` // 格式
Prompt string `json:"prompt"` // 提示词 Prompt string `json:"prompt"` // 提示词
NegativePrompt string `json:"negative_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) CfgScale int `json:"cfg_scale"` // 引导比例(minimum: 1; maximum: 20)
SamplerName string `json:"sampler_name"` // 采样器名称 SamplerName string `json:"sampler_name"` // 采样器名称
Seed int `json:"seed"` // 随机种子(minimum: 0; maximum: 2147483647) Seed int `json:"seed"` // 随机种子(minimum: 0; maximum: 2147483647)

2
models/Model.go
View File

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

2
routers/images.go
View File

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