【千帆SDK】利用千帆大模型平台实现文生图

原创
01/23 18:47
阅读数 21

💡学习前小提示 

请大家点击链接并加🌟:https://github.com/baidubce/bce-qianfan-sdk

千帆平台提供了各类大模型调用,包括文生文和文生图,千帆 SDK 提供了极为方便的方式进行调用,下述内容要求:

qianfan >= 0.2.8
# 安装依赖# install your dependency%pip install pillow qianfan

在使用千帆 SDK 前需要先准备自己的 Access Key 与 Secret Key 用作鉴权,可以从百度智能云控制台 - 安全认证处获取,详细流程可以参见文档

然后可以通过如下方式传递鉴权的 key
import osos.environ["QIANFAN_ACCESS_KEY"]="your ak"os.environ["QIANFAN_SECRET_KEY"]="your sk"

文生图

千帆平台预置了 Stable-Diffusion-XL 模型,可以提供文生图能力。
import qianfanfrom PIL import Image import io
t2i = qianfan.Text2Image()resp = t2i.do(prompt="A Ragdoll cat with a bowtie.", with_decode="base64")img_data = resp["body"]["data"][0]["image"]
img = Image.open(io.BytesIO(img_data))display(img)

千帆平台上还预置了大量 prompt 可以帮助您省去调优 prompt 的过程,快速生成高质量图片,千帆 SDK 也支持快速拉取平台预置 prompt

import qianfanfrom qianfan.common.hub import hub
p = hub.load("prompt/游戏原画1")prompt, neg_prompt = p.render()
resp = t2i.do(prompt=prompt, negative_prompt=neg_prompt, with_decode="base64")img_data = resp["body"]["data"][0]["image"]
img = Image.open(io.BytesIO(img_data))display(img)


本文分享自微信公众号 - 百度开发者中心(baidudev)。
如有侵权,请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”,欢迎正在阅读的你也加入,一起分享。

展开阅读全文
加载中
点击引领话题📣 发布并加入讨论🔥
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部