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

lora update

Browse Source
This commit is contained in:
桜華
2023-07-17 16:06:11 +08:00
parent 64707229a5
commit 0463196ad6
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
models/server.go
View File

@@ -94,13 +94,21 @@ func InitDefaultServer() (err error) {
// 初始化服务器中的模型列表 // 初始化服务器中的模型列表
func (server *Server) InitModels() (err error) { func (server *Server) InitModels() (err error) {
// 刷新检查点 /sdapi/v1/reload-checkpoint // 刷新检查点 /sdapi/v1/refresh-checkpoint
resp_update, err := http.Post(fmt.Sprintf("http://%s:%d/sdapi/v1/refresh-checkpoints", server.IP, server.Port), "application/json", nil) resp_update, err := http.Post(fmt.Sprintf("http://%s:%d/sdapi/v1/refresh-checkpoints", server.IP, server.Port), "application/json", nil)
if err != nil { if err != nil {
return fmt.Errorf("更新服务器中的模型列表失败: %v", err) return fmt.Errorf("更新服务器中的模型列表失败: %v", err)
} }
defer resp_update.Body.Close() defer resp_update.Body.Close()
log.Println("更新服务器中的模型列表:", resp_update.Status) log.Println("更新服务器中的ckpt模型列表:", resp_update.Status)
// 刷新检查点 /sdapi/v1/refresh-loras
lora_update, err := http.Post(fmt.Sprintf("http://%s:%d/sdapi/v1/refresh-loras", server.IP, server.Port), "application/json", nil)
if err != nil {
return fmt.Errorf("更新服务器中的模型列表失败: %v", err)
}
defer lora_update.Body.Close()
log.Println("更新服务器中的lora模型列表:", lora_update.Status)
// 获取服务器中的模型列表(Lora) // 获取服务器中的模型列表(Lora)
resp_lora, err := http.Get(fmt.Sprintf("http://%s:%d/sdapi/v1/loras", server.IP, server.Port)) resp_lora, err := http.Get(fmt.Sprintf("http://%s:%d/sdapi/v1/loras", server.IP, server.Port))