Stability AI 图像生成
Spring AI 支持 Stability AI 的 文本到图像生成模型。
先决条件
你需要创建一个 Stability AI 的 API 密钥来访问他们的 AI 模型,请遵循他们的 入门文档。
Spring AI 项目定义了一个名为 spring.ai.stabilityai.api-key
的配置属性,你应该将其设置为从 Stability AI 获取的 API Key
的值。导出环境变量是设置该配置属性的一种方式。
export SPRING_AI_STABILITYAI_API_KEY=<INSERT KEY HERE>
自动配置
Spring AI 为 Stability AI 图像生成客户端提供了 Spring Boot 自动配置。要启用它,请将以下依赖项添加到项目的 Maven pom.xml
文件中:
<dependency>
<groupId>org.springframework.ai</groupId>
<artifactId>spring-ai-stability-ai-spring-boot-starter</artifactId>
</dependency>
或者到你的 Gradle build.gradle
构建文件中。
dependencies {
implementation 'org.springframework.ai:spring-ai-stability-ai-spring-boot-starter'
}
请参考 依赖管理 部分,将 Spring AI BOM 添加到你的构建文件中。
图像生成属性
前缀 spring.ai.stabilityai
用作属性前缀,允许你连接到 Stability AI。
属性 | 描述 | 默认值 |
---|---|---|
spring.ai.stabilityai.base-url | 连接的 URL | api.stability.ai/v1 |
spring.ai.stabilityai.api-key | API 密钥 | - |
前缀 spring.ai.stabilityai.image
是用于配置 Stability AI 的 ImageModel
实现的属性前缀。
属性 | 描述 | 默认值 |
---|---|---|
spring.ai.stabilityai.image.enabled | 启用 Stability AI 图像模型。 | true |
spring.ai.stabilityai.image.base-url | 可选,覆盖 spring.ai.openai.base-url 以提供特定的 URL | [api.stability.ai/v1](https://api.stability.ai/v1) |
spring.ai.stabilityai.image.api-key | 可选,覆盖 spring.ai.openai.api-key 以提供特定的 API 密钥 | - |
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())
);
除了特定于模型的 StabilityAiImageOptions,你还可以使用一个可移植的 ImageOptions 实例,该实例通过 ImageOptionsBuilder#builder() 创建。