文心一格（精调后）
更新时间：2024-11-15
百度自主研发的跨模态图像生成大模型，创新知识增强的混合专家建模，是全球首个知识增强的AI作画大模型，在语义控制、图像细节、中文理解等方面优势显著，已作为基础模型应用在 文心一格 等相关业务上。本文介绍了经过SFT模型训练后发布的服务，调用相关API说明。
功能介绍
调用本接口，根据用户输入的文本生成图片。
注意事项
- 调用本文API前，需先经过SFT模型训练后发布服务，相关内容请参考新手指南-平台使用快速开始。
- 支持通过Python SDK、Go SDK、Java SDK和Node.js SDK调用，调用流程请查看SDK安装及使用流程。
SDK调用
调用示例
对于用户自行发布的模型服务，如文心一格（精调后），通过使用endpoint进行调用，示例如下。
import os
import qianfan
from PIL import Image
import io
# 使用安全认证AK/SK鉴权，通过环境变量方式初始化；替换下列示例中参数，安全认证Access Key替换your_iam_ak，Secret Key替换your_iam_sk
os.environ["QIANFAN_ACCESS_KEY"] = "your_iam_ak"
os.environ["QIANFAN_SECRET_KEY"] = "your_iam_sk"
t2i = qianfan.Text2Image()
resp = t2i.do(prompt="A Ragdoll cat with a bowtie.", with_decode="base64",endpoint="your_custom_endpoint")
img_data = resp["body"]["data"][0]["image"]
img = Image.open(io.BytesIO(img_data))
img.show(img)import com.baidubce.qianfan.Qianfan;
import com.baidubce.qianfan.model.image.Text2ImageResponse;
public class Main {
public static void main(String[] args) {
// 使用安全认证AK/SK鉴权，替换下列示例中参数，安全认证Access Key替换your_iam_ak，Secret Key替换your_iam_sk
Qianfan qianfan = new Qianfan("your_iam_ak", "your_iam_sk");
// 调用文生图模型
Text2ImageResponse response = qianfan.text2Image().endpoint("your_custom_endpoint")
.prompt("A Ragdoll cat with a bowtie")
.execute();
// 返回信息，返回图片base64编码信息等
System.out.println("created: "+response.getCreated());
System.out.println("id: "+response.getId());
System.out.println("prompttokens: "+response.getUsage().getPromptTokens());
System.out.println("totaltokens: "+response.getUsage().getTotalTokens());
System.out.println("b64image: "+response.getData().getFirst().getB64Image());
System.out.println("image: "+ Arrays.toString(response.getData().getFirst().getImage()));
System.out.println("index: "+response.getData().getFirst().getIndex());
System.out.println("object: "+response.getObject());
}package main
import (
"context"
"encoding/base64"
"fmt"
"os"
"github.com/baidubce/bce-qianfan-sdk/go/qianfan"
)
func main() {
// 使用安全认证AK/SK鉴权，通过环境变量初始化；替换下列示例中参数，安全认证Access Key替换your_iam_ak，Secret Key替换your_iam_sk
os.Setenv("QIANFAN_ACCESS_KEY", "your_iam_ak")
os.Setenv("QIANFAN_SECRET_KEY", "your_iam_sk")
text2img := qianfan.NewText2Image(
qianfan.WithEndpoint("your_custom_endpoint"),
)
resp, err := text2img.Do(context.TODO(), &qianfan.Text2ImageRequest{
Prompt: "A Ragdoll cat with a bowtie.",
})
if err != nil {
panic(err)
}
img := resp.Data[0].Base64Image
imgData, err := base64.StdEncoding.DecodeString(img)
if err != nil {
panic(err)
}
outputFile := "output.jpg"
f, err := os.Create(outputFile)
if err != nil {
panic(err)
}
defer func() {
if err := f.Close(); err != nil {
panic(err)
}
}()
_, err = f.Write(imgData)
if err != nil {
panic(err)
}
fmt.Printf("Image saved to %s\n", outputFile)
}