Stability AI 图像生成

本站(springdoc.cn)中的内容来源于 spring.io ,原始版权归属于 spring.io。由 springdoc.cn 进行翻译,整理。可供个人学习、研究,未经许可,不得进行任何转载、商用或与之相关的行为。 商标声明:Spring 是 Pivotal Software, Inc. 在美国以及其他国家的商标。

Spring AI 支持 Stability AI 的 文生图模型(Text-to-Image Generation Model)

先决条件

你需要向 Stability AI 创建一个 API 密钥,才能访问他们的 AI 模型。请按照他们的 入门文档 获取 API 密钥。

Spring AI 项目定义了一个名为 spring.ai.stabilityai.api-key 的配置属性,你应将其设置为从 Stability AI 获取的 API Key 的值。

你可以在 application.properties 文件中设置该配置属性:

spring.ai.stabilityai.api-key=<your-stabilityai-api-key>

在处理 API Key 等敏感信息时,为了增强安全性,可以使用 Spring 表达式语言(SpEL)引用自定义环境变量:

# In application.yml
spring:
  ai:
    stabilityai:
      api-key: ${STABILITYAI_API_KEY}
# In your environment or .env file
export STABILITYAI_API_KEY=<your-stabilityai-api-key>

你也可以在应用代码中以编程方式设置该配置:

// Retrieve API key from a secure source or environment variable
String apiKey = System.getenv("STABILITYAI_API_KEY");

自动配置

Spring AI 自动配置、Starter 模块的工件名称发生了重大变化。更多信息请参阅 升级说明

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 BOM 添加到构建文件中。

图像生成配置

spring.ai.stabilityai 作为属性前缀,可让你连接到 Stability AI。

属性 说明 默认值

spring.ai.stabilityai.base-url

要连接到的 URL

api.stability.ai/v1

spring.ai.stabilityai.api-key

API Key

-

图像自动配置的启用和禁用现在可通过以 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

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/jsonimage/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 之外,你还可以使用通过 ImageOptionsBuilder#builder() 创建的便携式 ImageOptions 实例。