From 320f7ff542f36fd1fe9d1b8fbaa52ce3e75876d4 Mon Sep 17 00:00:00 2001
From: watany <76135106+watany-dev@users.noreply.github.com>
Date: Thu, 22 Feb 2024 04:33:19 +0000
Subject: [PATCH] chore(bedrock): stabilityai model

---
 packages/aws-cdk-lib/aws-bedrock/lib/foundation-model.ts | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/packages/aws-cdk-lib/aws-bedrock/lib/foundation-model.ts b/packages/aws-cdk-lib/aws-bedrock/lib/foundation-model.ts
index b4e0a31d9c7fc..645a696aa7d56 100644
--- a/packages/aws-cdk-lib/aws-bedrock/lib/foundation-model.ts
+++ b/packages/aws-cdk-lib/aws-bedrock/lib/foundation-model.ts
@@ -5,7 +5,7 @@ import { ArnFormat, Stack } from '../../core';
 /**
  * The model identifiers for the Bedrock base foundation models.
  *
- * @see https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids-arns.html
+ * @see https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids.html
  */
 export class FoundationModelIdentifier {
   /** Base model "ai21.j2-mid-v1". */
@@ -56,6 +56,12 @@ export class FoundationModelIdentifier {
   /** Base model "meta.llama2-70b-chat-v1". */
   public static readonly META_LLAMA_2_CHAT_70B_V1 = new FoundationModelIdentifier('meta.llama2-70b-chat-v1');
 
+  /** Base model "stability.stable-diffusion-xl-v0". */
+  public static readonly STABILITY_STABLE_DIFFUSION_XL_V0 = new FoundationModelIdentifier('stability.stable-diffusion-xl-v0');
+
+  /** Base model "stability.stable-diffusion-xl-v1". */
+  public static readonly STABILITY_STABLE_DIFFUSION_XL_V1 = new FoundationModelIdentifier('stability.stable-diffusion-xl-v1');
+
   /**
    * Constructor for foundation model identifier
    * @param modelId the model identifier