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

12
routers/images.go
View File

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