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

提示词引导系数 (CFG Scale)

Browse Source
This commit is contained in:
桜華
2023-06-23 02:06:05 +08:00
parent 546ff57d6a
commit 8e874f7d0f
3 changed files with 21 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
models/Image.go
View File

@@ -28,7 +28,7 @@ type Image struct {
Prompt string `json:"prompt"` // 提示词
NegativePrompt string `json:"negative_prompt"` // 负向提示
NumInferenceSteps int `json:"num_inference_steps"` // 推理步数(minimum: 1; maximum: 500)
GuidanceScale float32 `json:"guidance_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)
Seed string `json:"seed"` // 随机种子(minimum: 0; maximum: 2147483647)
FromImage int `json:"from_image"` // 来源图片(如果是从图片生成的, 则记录来源图片的ID)

12
models/Model.go
View File

@@ -108,9 +108,9 @@ func (model *Model) Inference(image_list []Image, callback func(Image)) {
//SeedResizeFromW int `json:"seed_resize_from_w"`
//SamplerName string `json:"sampler_name"`
//BatchSize int `json:"batch_size"`
NIter int `json:"n_iter"`
Steps int `json:"steps"`
//CfgScale int `json:"cfg_scale"`
NIter int `json:"n_iter"`
Steps int `json:"steps"`
CfgScale int `json:"cfg_scale"`
//Width int `json:"width"`
//Height int `json:"height"`
//RestoreFaces bool `json:"restore_faces"`
@@ -154,9 +154,9 @@ func (model *Model) Inference(image_list []Image, callback func(Image)) {
//SeedResizeFromW: -1,
//SamplerName: "beamsearch",
//BatchSize: 1,
NIter: len(image_list),
Steps: 50,
//CfgScale: 7,
NIter: len(image_list), // 1~100
Steps: 50, // 1~150
CfgScale: image_list[0].CfgScale,
//Width: 512,
//Height: 512,
//RestoreFaces: false,