Stability AI 影像生成

Spring AI 支援 Stability AI 的 文字到影像生成模型

先決條件

您需要使用 Stability AI 建立 API 金鑰才能存取他們的 AI 模型,請依照他們的 入門文件 進行操作。

Spring AI 專案定義了一個名為 spring.ai.stabilityai.api-key 的組態屬性,您應該將其設定為從 Stability AI 取得的 API 金鑰 值。匯出環境變數是設定該組態屬性的一種方式。

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

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()
                .withStylePreset("cinematic")
                .withN(4)
                .withHeight(1024)
                .withWidth(1024).build())

);
除了模型特定的 StabilityAiImageOptions,您可以使用可攜式的 ImageOptions 實例,使用 ImageOptionsBuilder#builder() 建立。