logo
4
话题头图

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

💡学习前小提示
请大家点击链接并加🌟: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 os
os.environ["QIANFAN_ACCESS_KEY"]="your ak"
os.environ["QIANFAN_SECRET_KEY"]="your sk"

文生图

千帆平台预置了 Stable-Diffusion-XL 模型,可以提供文生图能力。
  
  
  
  
  
  
import qianfan
from 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 qianfan
from 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)
评论
用户头像