模型超时

models/Model.go

@@ -113,6 +113,9 @@ func (model *Model) Inference(image_list []Image, callback func(Image)) {
 			SdModelCheckpoint string `json:"sd_model_checkpoint"`
 		}{}
 
+		// 超时时间 1分钟
+		var timeout = time.Now().Add(time.Second * 60)
+
 		for {
 			if err := goreq.Get(fmt.Sprintf("http://%s:%d/sdapi/v1/options", server.IP, server.Port)).Do().BindJSON(&form); err != nil {
 				log.Println("获取推理机配置失败:", err)
@@ -122,6 +125,10 @@ func (model *Model) Inference(image_list []Image, callback func(Image)) {
 				log.Println("模型切换完成:", form.SdModelCheckpoint)
 				break
 			}
+			if time.Now().After(timeout) {
+				log.Println("模型切换超时:", form.SdModelCheckpoint)
+				return
+			}
 			time.Sleep(time.Second)
 		}