36 lines
1.1 KiB
Go
36 lines
1.1 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))
|
|
}
|
|
|
|
func ParamsModelsGet(w http.ResponseWriter, r *http.Request) {
|
|
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"] = []string{"512x512", "768x768", "1024x768"}
|
|
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))
|
|
}
|