Stability AI 图像生成
Spring AI 支持 Stability AI 的 文本生成图像模型。
前置条件
你需要创建一个 Stability AI 的 API 密钥才能访问他们的 AI 模型。请参考 Stability AI 的 入门文档 获取 API 密钥。
Spring AI 项目定义了一个配置属性 spring.ai.stabilityai.api-key,你应将其设置为从 Stability AI 获取的 API 密钥值。
可以在 application.properties 文件中设置此配置属性:
spring.ai.stabilityai.api-key=<your-stabilityai-api-key>
为了在处理敏感信息(如 API 密钥)时增强安全性,你可以使用 Spring 表达式语言(SpEL)引用自定义环境变量:
# 在 application.yml 中
spring:
ai:
stabilityai:
api-key: ${STABILITYAI_API_KEY}
# 在你的环境或 .env 文件中
export STABILITYAI_API_KEY=<your-stabilityai-api-key>
你也可以在应用代码中以编程方式设置此配置:
// 从安全来源或环境变量中获取 API 密钥
String apiKey = System.getenv("STABILITYAI_API_KEY");
自动配置
Spring AI 为 Stability AI 图像生成客户端提供了 Spring Boot 自动配置。要启用该功能,请将以下依赖添加到项目的 Maven pom.xml 文件中:
<dependency>
<groupId>org.springframework.ai</groupId>
<artifactId>spring-ai-starter-model-stability-ai</artifactId>
</dependency>
或者添加到 Gradle build.gradle 文件中:
dependencies {
implementation 'org.springframework.ai:spring-ai-starter-model-stability-ai'
}
图像生成属性
spring.ai.stabilityai 是用于连接 Stability AI 的配置属性前缀。
| 属性 | 描述 | 默认值 |
|---|---|---|
spring.ai.stabilityai.base-url | 连接的 URL | api.stability.ai/v1 |
spring.ai.stabilityai.api-key | API 密钥 | - |
spring.ai.model.image 进行配置:- 启用:
spring.ai.model.image=stabilityai(默认启用) - 禁用:
spring.ai.model.image=none(或任何不匹配stabilityai的值)
spring.ai.stabilityai.image 是用于配置 Stability AI ImageModel 实现的属性前缀。
| 属性 | 描述 | 默认值 |
|---|---|---|
spring.ai.stabilityai.image.enabled | 启用 Stability AI 图像模型(已移除,不再有效) | true |
spring.ai.model.image | 启用 Stability AI 图像模型 | stabilityai |
spring.ai.stabilityai.image.base-url | 可选,覆盖 spring.ai.openai.base-url 提供特定 URL | https://api.stability.ai/v1 |
spring.ai.stabilityai.image.api-key | 可选,覆盖 spring.ai.openai.api-key 提供特定 API Key | - |
spring.ai.stabilityai.image.option.n | 要生成的图像数量,必须在 1~10 之间 | 1 |
spring.ai.stabilityai.image.option.model | 使用的 Stability AI 引擎/模型,模型通过 URL 作为路径参数传入 | stable-diffusion-v1-6 |
spring.ai.stabilityai.image.option.width | 生成图像的宽度(像素),需为 64 的倍数 | 512 |
spring.ai.stabilityai.image.option.height | 生成图像的高度(像素),需为 64 的倍数 | 512 |
spring.ai.stabilityai.image.option.responseFormat | 返回生成图像的格式,可为 "application/json" 或 "image/png" | - |
spring.ai.stabilityai.image.option.cfg_scale | 扩散过程严格遵循提示文本的程度,范围 0~35 | 7 |
spring.ai.stabilityai.image.option.clip_guidance_preset | 用于引导图像模型生成特定风格的样式预设 | NONE |
spring.ai.stabilityai.image.option.sampler | 扩散过程使用的采样器,若未设置则自动选择 | - |
spring.ai.stabilityai.image.option.seed | 随机噪声种子(可省略或使用 0 随机生成),有效范围 0~4294967295 | 0 |
spring.ai.stabilityai.image.option.steps | 扩散步骤数,范围 10~50 | 30 |
spring.ai.stabilityai.image.option.style_preset | 用于引导图像模型生成特定风格的样式预设 | - |
运行时选项
StabilityAiImageOptions.java 提供了模型配置选项,例如使用的模型、风格、尺寸等。
在应用启动时,可以通过 StabilityAiImageModel(StabilityAiApi stabilityAiApi, StabilityAiImageOptions options) 构造函数配置默认选项。或者,也可以使用之前介绍的 spring.ai.openai.image.options.* 属性进行配置。
在运行时,你可以通过在 ImagePrompt 调用中添加特定请求的选项来覆盖默认配置。例如,要覆盖 Stability AI 特定的选项,如图像质量和生成图像的数量,可以使用如下代码示例:
ImageResponse response = stabilityaiImageModel.call(
new ImagePrompt("A light cream colored mini golden doodle",
StabilityAiImageOptions.builder()
.stylePreset("cinematic")
.N(4)
.height(1024)
.width(1024)
.build())
);