From 0463196ad6fc807be685233f61f42a999d35e79f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= Date: Mon, 17 Jul 2023 16:06:11 +0800 Subject: [PATCH] lora update --- models/server.go | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/models/server.go b/models/server.go index 7601340..c590e94 100644 --- a/models/server.go +++ b/models/server.go @@ -94,13 +94,21 @@ func InitDefaultServer() (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) if err != nil { return fmt.Errorf("更新服务器中的模型列表失败: %v", err) } 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) resp_lora, err := http.Get(fmt.Sprintf("http://%s:%d/sdapi/v1/loras", server.IP, server.Port))