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 18:39:21 +08:00
parent 5925625c3f
commit 01b16f8926
1 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
routers/params.go
View File

@@ -23,12 +23,28 @@ func ParamsListGet(w http.ResponseWriter, r *http.Request) {
w.Write(utils.ToJSON(listview)) w.Write(utils.ToJSON(listview))
} }
type Size struct {
Width int `json:"width"`
Height int `json:"height"`
AspectRatio string `json:"aspect_ratio"`
}
func ParamsModelsGet(w http.ResponseWriter, r *http.Request) { func ParamsModelsGet(w http.ResponseWriter, r *http.Request) {
var size_list []Size
size_list = append(size_list, Size{Width: 512, Height: 512, AspectRatio: "1:1"})
size_list = append(size_list, Size{Width: 768, Height: 768, AspectRatio: "1:1"})
size_list = append(size_list, Size{Width: 1024, Height: 768, AspectRatio: "4:3"})
size_list = append(size_list, Size{Width: 1024, Height: 1024, AspectRatio: "1:1"})
size_list = append(size_list, Size{Width: 1280, Height: 720, AspectRatio: "16:9"})
size_list = append(size_list, Size{Width: 1280, Height: 1024, AspectRatio: "5:4"})
size_list = append(size_list, Size{Width: 1920, Height: 1080, AspectRatio: "16:9"})
size_list = append(size_list, Size{Width: 1920, Height: 1200, AspectRatio: "16:10"})
params := make(map[string]interface{}) params := make(map[string]interface{})
params["type"] = []string{"lora", "ckp", "hyper", "ti"} params["type"] = []string{"lora", "ckp", "hyper", "ti"}
params["status"] = []string{"pending", "running", "finished", "failed"} params["status"] = []string{"pending", "running", "finished", "failed"}
params["base_model"] = []string{"SD1.5", "SD2"} params["base_model"] = []string{"SD1.5", "SD2"}
params["size"] = []string{"512x512", "768x768", "1024x768"} params["size"] = size_list
params["sampler_name"] = []string{"Euler a", "Euler", "LMS", "Heun", "DPM2", "DPM2 a", "DPM++ 2S a", "DPM++ 2M", "DPM++ SDE", "DPM++ 2M SDE", "DPM fast", "DDIM"} params["sampler_name"] = []string{"Euler a", "Euler", "LMS", "Heun", "DPM2", "DPM2 a", "DPM++ 2S a", "DPM++ 2M", "DPM++ SDE", "DPM++ 2M SDE", "DPM fast", "DDIM"}
w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.Write(utils.ToJSON(params)) w.Write(utils.ToJSON(params))