fix: chinese prompt for sd3, add sd3 price

aws-samples · Jan 3, 2025 · 3497bb0 · 3497bb0
1 parent 7feedb2
commit 3497bb0
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/react-native/src/chat/component/ImageProgressBar.tsx b/react-native/src/chat/component/ImageProgressBar.tsx
@@ -77,7 +77,6 @@ const ImageProgressBar = () => {
         useNativeDriver: true,
       }).start();
     } else if (event?.event === 'onImageStop') {
-      console.log('onImageStop');
       startProgressAnimation(100, 200).start();
     } else if (event?.event === 'onImageStart') {
       Animated.sequence([

diff --git a/react-native/src/settings/ModelPrice.ts b/react-native/src/settings/ModelPrice.ts
@@ -241,6 +241,9 @@ export const ModelPrice: ModelPriceType = {
     'SD3 Large 1.0': {
       medium: 0.08,
     },
+    'Stable Diffusion 3.5 Large': {
+      medium: 0.08,
+    },
     'Stable Image Core 1.0': {
       medium: 0.04,
     },
@@ -277,6 +280,9 @@ type ImageModelPrices = {
   'SD3 Large 1.0': {
     medium: number;
   };
+  'Stable Diffusion 3.5 Large': {
+    medium: number;
+  };
   'Stable Image Core 1.0': {
     medium: number;
   };

diff --git a/server/src/main.py b/server/src/main.py
@@ -141,7 +141,7 @@ async def gen_image(request: ImageRequest,
     region = request.region
     client = boto3.client("bedrock-runtime",
                           region_name=region)
-    if ref_images is None and contains_chinese(prompt):
+    if (ref_images is None or model_id.startswith("stability.")) and contains_chinese(prompt):
         prompt = get_english_prompt(client, prompt)
     return get_image(client, model_id, prompt, ref_images, width, height)