使用Amazon Bedrock的API进行文生图的实验

原创

已于 2024-04-24 14:53:40 修改 · 1.2k 阅读

于 2024-04-24 10:34:54 首次发布

因为最近GenAI相当火爆，正好参与了Amazon Bedrock相关的项目，趁这个机会对Amazon Bedrock的学习分享一下。

1. 开通本账号下Bedrock相应模型的使用权限

登录AWS Global账号后，选择服务->Machine Learning->Amazon Bedrock
在这里插入图片描述

在Amazon Bedrock页面，点击入门

在这里插入图片描述

在左侧导航栏中，选择模型访问权限
在这里插入图片描述

在模型访问权限页面，点击管理模型访问权限

在这里插入图片描述

选择需要使用的模型，点击保存更改
在这里插入图片描述

Anthropic的Claude模型需要填写应用场景描述

当相应的模型后面显示已授权访问权限后，方可使用

在这里插入图片描述

2. 在AWS portal中进行文生图的测试

在Amazon Bedrock页面，点击图像

在这里插入图片描述

在选择模型页面，选择Stability AI -> SDXL 1.0 -> 应用

在这里插入图片描述

在提示词中输入：

Emma Watson as a powerful mysterious sorceress, casting lightning magic, detailed clothing, digital painting, hyperrealistic, fantasy, Surrealist, full body, by Stanley Artgerm Lau and Alphonse Mucha, artstation

其他配置如下图：

在这里插入图片描述

操作解释：参考

操作	内容
模式	模型生成新图像（生成）或编辑（编辑）在参考图像中提供的图像
否定提示	不希望模型生成的项目或概念，例如卡通或暴力
推理图像	用于生成响应的图像或希望模型编辑的图像
响应图像	生成图像的输出设置，例如质量、方向、大小和要生成的图像数量
方向	用于生成图像的方向
大小	用于生成图像的大小
高级配置	要传递给模型的推理参数

具体其他配置请参考: Amazon Bedrock操场

3. 在代码中调用Bedrock API，部署文生图应用

部署一个EC2，参考

登录该虚拟机，安装python等必要环境

在这里插入图片描述

输入相应权限AKSK

在这里插入图片描述

创建一个目录

mkdir genai && cd genai

创建一个python文件

vi genai.py

粘贴下面这个代码

import streamlit as st
from PIL import Image
import base64
import boto3
import io
import json
import os

DEBUG = os.getenv("DEBUG", False)
DEFAULT_SEED = os.getenv("DEFAULT_SEED", 12345)
MAX_SEED = 4294967295
MODEL_ID = "stability.stable-diffusion-xl-v1"
NEGATIVE_PROMPTS = [
    "bad anatomy", "distorted", "blurry",
    "pixelated", "dull", "unclear",
    "poorly rendered",
    "poorly Rendered face",
    "poorly drawn face",
    "poor facial details",
    "poorly drawn hands",
    "poorly rendered hands",