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

使用默认参数推理

Browse Source
This commit is contained in:
桜華
2023-06-21 16:13:52 +08:00
parent 2a71384fad
commit 1d0da612b8
4 changed files with 40 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
routers/images.go
View File

@@ -154,6 +154,11 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) {
return
}
// 推理图像
go model.Inference([]models.Image{}, func() {
log.Println("推理完成")
})
// 直接创建一组图片
var image_list []models.Image
var task string = uuid.New().String()

24
routers/servers.go
View File

@@ -26,18 +26,18 @@ func ServersGet(w http.ResponseWriter, r *http.Request) {
for _, server := range server_list {
server.CheckStatus() // 驗證服務器狀態
//// 讀取模型信息
resp, err := http.Get(fmt.Sprintf("http://%s:%d/sdapi/v1/sd-models", server.IP, server.Port))
if err != nil || resp.StatusCode != http.StatusOK {
server.Models = []map[string]interface{}{}
} else {
var models []map[string]interface{}
body, _ := ioutil.ReadAll(resp.Body)
defer resp.Body.Close()
if err := json.Unmarshal(body, &models); err != nil {
server.Models = []map[string]interface{}{}
}
server.Models = models
}
//resp, err := http.Get(fmt.Sprintf("http://%s:%d/sdapi/v1/sd-models", server.IP, server.Port))
//if err != nil || resp.StatusCode != http.StatusOK {
// server.Models = []map[string]interface{}{}
//} else {
// var models []map[string]interface{}
// body, _ := ioutil.ReadAll(resp.Body)
// defer resp.Body.Close()
// if err := json.Unmarshal(body, &models); err != nil {
// server.Models = []map[string]interface{}{}
// }
// server.Models = models
//}
listview.List = append(listview.List, server)
}
listview.Next = listview.Page*listview.PageSize < int(listview.Total)