访问官网
Github
文档
论文
SD-WEBUI-GO
Golang SDK for stable-diffusion-webui 的 API
贡献者中有 AUTOMATIC1111，这是我的一个小玩笑😂。
在此，诚挚感谢 AUTOMATIC1111 对 AIGC 的巨大贡献
这是一个基于 stable-diffusion-webui 的 Go 语言版本 SDK。在您的代码中，可以通过面向对象的操作直接使用 stable-diffusion-webui 的 API 接口，而不需要处理繁琐的 JSON。
支持扩展 API ！
您可以在这个 wiki 页面 查看 intersvc 的支持列表。
使用方法
有两种方法可以使用这个 SDK。
-
使用 'intersvc'，它为 'sd-webui' 提供了高度封装的 API 接口。使用此方法感觉像是在直接使用 'sd-webui' 的 Go 语言版本。
-
使用 'go-swagger'，虽然仍然涉及面向对象的操作，但比 'intersvc' 略复杂。
第二种方法几乎支持所有接口，而第一种方法正逐步得到支持。
实际上，大部分接口可以通过 'intersvc' 使用，但需要定义响应。'sd-webui' 的 API 文档没有提供响应的内容，所以需要手动定义。您可以参考下面的 "参与" 部分的具体说明。
go-swagger 的 text2img
package main
import (
"encoding/base64"
"fmt"
SdClient "github.com/SpenserCai/sd-webui-go"
SdApiOperation "github.com/SpenserCai/sd-webui-go/stablediffusion/client/operations"
SdApiModel "github.com/SpenserCai/sd-webui-go/stablediffusion/models"
)
func MustBeNil(err error) {
if err != nil {
panic(err)
}
}
var URL = "127.0.0.1:7860"
func main() {
sdClient := SdClient.NewStableDiffInterface(URL)
rd := SdApiOperation.NewText2imgapiSdapiV1Txt2imgPostParams()
rd.Body = &SdApiModel.StableDiffusionProcessingTxt2Img{
Prompt: "dog",
NegativePrompt: "ugly",
ScriptArgs: []interface{}{},
}
resp, err := sdClient.Client.Operations.Text2imgapiSdapiV1Txt2imgPost(rd)
MustBeNil(err)
for i, s := range resp.Payload.Images {
b, err := base64.StdEncoding.DecodeString(s)
MustBeNil(err)
err = os.WriteFile(fmt.Sprintf("%d.png", i), b, 0644)
MustBeNil(err)
}
}
intersvc 的 deoldify
import (
SdClient "github.com/SpenserCai/sd-webui-go"
"github.com/SpenserCai/sd-webui-go/intersvc"
)
func main() {
// 创建客户端
sdClient := SdClient.NewStableDiffInterface("127.0.0.1:7860")
var f_factor int64 = 20
var artistic bool = false
// 设置请求
deoldify_inter := &intersvc.DeoldifyImage{
RequestItem: &intersvc.DeoldifyImageRequest{
InputImage: "https://media.discordapp.net/attachments/1138408545277190237/1138508881635577947i7krs1njekla1.jpg",
RenderFactor: &f_factor,
Artistic: &artistic,
},
}
deoldify_inter.Action(sdClient)
if deoldify_inter.Error != nil {
panic(deoldify_inter.Error)
}
response := deoldify_inter.GetResponse()
}
完整示例代码: intersvc_example
go-swagger 的 deoldify
import (
"encoding/base64"
"os"
SdClient "github.com/SpenserCai/sd-webui-go"
intersvc "github.com/SpenserCai/sd-webui-go/intersvc"
SdApiOperation "github.com/SpenserCai/sd-webui-go/stablediffusion/client/operations"
SdApiModel "github.com/SpenserCai/sd-webui-go/stablediffusion/models"
)
type DeoldifyResponse struct {
Image string `json:"image"`
}
func main() {
var (
err error
)
// 初始化客户端
sdClient := SdClient.NewStableDiffInterface("127.0.0.1:7860")
var f_factor int64 = 20
var artistic bool = false
// 设置请求数据
RequestData := SdApiOperation.NewDeoldifyImageDeoldifyImagePostParams()
RequestData.Body = &SdApiModel.BodyDeoldifyImageDeoldifyImagePost{
InputImage: "https://media.discordapp.net/attachments/1138408545277190237/1138508881635577947/i7krs1njekla1.jpg",
RenderFactor: &f_factor,
Artistic: &artistic,
}
// 发送请求
Response, err := sdClient.Client.Operations.DeoldifyImageDeoldifyImagePost(RequestData)
if err != nil {
panic(err)
}
// 转换响应
deoldifyRep, err := intersvc.ConvertResponse(Response.Payload, &DeoldifyResponse{})
if err != nil {
panic(err)
}
response := deoldifyRep.(*DeoldifyResponse)
}
完整示例代码: go-swagger_example
参与
大部分 intersvc 的代码是使用代码生成器生成的。但是，由于 sd-webui 的 API 文档中缺乏响应信息，这些需要手动编写。
如何提交 PR
您需要 fork 当前代码库的 dev 分支，进行必要的代码更新，在自己的仓库中创建一个名为 dev-[model filename] 的分支，然后提交一个 pull request 到当前代码库的 dev 分支。
如何定义响应模型
在 sd-webui-go/intersvc 中您可以看到一些像 ***_model.go 的文件。这些文件定义了响应模型。您可以参照以下示例来定义响应模型。
从
type DeoldifyImageResponse struct {
}
到
type DeoldifyImageResponse struct {
Image string `json:"image"`
}
