Merge pull request #54 from SpenserCai/dev

change time out
SpenserCai · Aug 26, 2023 · bafb8b0 · bafb8b0
2 parents d6a40f5 + 11c9a6a
commit bafb8b0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/webui.go b/webui.go
@@ -3,12 +3,16 @@
  * @Date: 2023-08-11 13:15:25
  * @version:
  * @LastEditors: SpenserCai
- * @LastEditTime: 2023-08-12 01:08:28
+ * @LastEditTime: 2023-08-26 12:35:45
  * @Description: file content
  */
 package webui
 
 import (
+	"time"
+
+	rclient "github.com/go-openapi/runtime/client"
+
 	StableDiffClient "github.com/SpenserCai/sd-webui-go/stablediffusion/client"
 
 	"github.com/go-openapi/strfmt"
@@ -19,6 +23,7 @@ type StableDiffInterface struct {
 }
 
 func NewStableDiffInterface(host string) *StableDiffInterface {
+	rclient.DefaultTimeout = 120 * time.Second
 	var client *StableDiffClient.StableDiffusion
 	if host == "" {
 		client = StableDiffClient.NewHTTPClient(strfmt.Default)