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 02:45:53 +08:00
parent fa2102af38
commit ec33c9fafd
3 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
models/Image.go
View File

@@ -29,7 +29,7 @@ type Image struct {
NegativePrompt string `json:"negative_prompt"` // 负向提示 NegativePrompt string `json:"negative_prompt"` // 负向提示
NumInferenceSteps int `json:"num_inference_steps"` // 推理步数(minimum: 1; maximum: 500) NumInferenceSteps int `json:"num_inference_steps"` // 推理步数(minimum: 1; maximum: 500)
CfgScale int `json:"cfg_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) SamplerName string `json:"sampler_name"` // 采样器名称
Seed int `json:"seed"` // 随机种子(minimum: 0; maximum: 2147483647) Seed int `json:"seed"` // 随机种子(minimum: 0; maximum: 2147483647)
FromImage int `json:"from_image"` // 来源图片(如果是从图片生成的, 则记录来源图片的ID) FromImage int `json:"from_image"` // 来源图片(如果是从图片生成的, 则记录来源图片的ID)
Task string `json:"task"` // 任务编号(uuid) Task string `json:"task"` // 任务编号(uuid)

4
models/Model.go
View File

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

12
routers/images.go
View File

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