52 lines
1.8 KiB
Go
52 lines
1.8 KiB
Go
package routers
|
|
|
|
import (
|
|
"main/models"
|
|
"main/utils"
|
|
"net/http"
|
|
)
|
|
|
|
func ParamsListGet(w http.ResponseWriter, r *http.Request) {
|
|
var listview models.ListView
|
|
|
|
var item = make(map[string]interface{})
|
|
item["id"] = "model"
|
|
item["name"] = "模型"
|
|
|
|
listview.List = append(listview.List, item)
|
|
listview.Total = 1
|
|
listview.Page = 1
|
|
listview.PageSize = 10
|
|
listview.Next = false
|
|
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
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) {
|
|
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["type"] = []string{"lora", "ckp", "hyper", "ti"}
|
|
params["status"] = []string{"pending", "running", "finished", "failed"}
|
|
params["base_model"] = []string{"SD1.5", "SD2"}
|
|
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"}
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
w.Write(utils.ToJSON(params))
|
|
}
|