fix: update instance type for workshop

awslabs · Dec 19, 2023 · 96a175e · 96a175e
1 parent ba9080d
commit 96a175e
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/src/api-deployment/ai-solution-kit-workshop-stack.ts b/src/api-deployment/ai-solution-kit-workshop-stack.ts
@@ -171,6 +171,7 @@ export class AISolutionKitStack extends Stack {
         ecrDeployment: ecrDeployment,
         updateCustomResourceProvider: updateCustomResourceProvider,
         ecrRegistry: props?.ecrRegistry,
+        instanceType: 'ml.g5.xlarge',
       });
       (advancedOCRSageMaker.nestedStackResource as CfnStack).cfnOptions.condition = cfnTemplate.getCondition('ConditionAdvancedOCRSageMaker');
       this.addOutput(cfnTemplate, api.restApiId, 'advanced-ocr-ml', 'Advanced OCR - Multilingual', 'ConditionAdvancedOCRSageMaker');
@@ -184,6 +185,7 @@ export class AISolutionKitStack extends Stack {
         ecrDeployment: ecrDeployment,
         updateCustomResourceProvider: updateCustomResourceProvider,
         ecrRegistry: props.ecrRegistry,
+        instanceType: 'ml.g5.xlarge',
       });
       (customOCRSageMakerFeatureNestedStack.nestedStackResource as CfnStack).cfnOptions.condition = cfnTemplate.getCondition('ConditionCustomOCRSageMaker');
       this.addOutput(cfnTemplate, api.restApiId, 'custom-ocr-ml', 'Custom OCR SageMaker', 'ConditionCustomOCRSageMaker');
@@ -211,6 +213,7 @@ export class AISolutionKitStack extends Stack {
         ecrDeployment: ecrDeployment,
         updateCustomResourceProvider: updateCustomResourceProvider,
         ecrRegistry: props.ecrRegistry,
+        instanceType: 'ml.g5.xlarge',
       });
       (generalNLUSageMaker.nestedStackResource as CfnStack).cfnOptions.condition = cfnTemplate.getCondition('ConditionGeneralNLUSageMaker');
       this.addOutput(cfnTemplate, api.restApiId, 'general-nlu-ml', 'General NLU SageMaker', 'ConditionGeneralNLUSageMaker');

diff --git a/src/api-deployment/features/sagemaker/advanced-ocr-sagemaker.ts b/src/api-deployment/features/sagemaker/advanced-ocr-sagemaker.ts
@@ -19,7 +19,7 @@ export class AdvancedOCRSageMakerFeatureNestedStack extends FeatureNestedStack {
       featureName: `${featureName}`,
       featureCategory: 'media',
       updateCustomResourceProvider: props.updateCustomResourceProvider,
-      sageMakerInstanceType: 'ml.g4dn.xlarge',
+      sageMakerInstanceType: props.instanceType == null ? 'ml.g4dn.xlarge' : props.instanceType,
       sageMakerEcrDeployment: props.ecrDeployment,
     });
   }

diff --git a/src/api-deployment/features/sagemaker/custom-ocr-sagemaker.ts b/src/api-deployment/features/sagemaker/custom-ocr-sagemaker.ts
@@ -19,7 +19,7 @@ export class CustomOCRSageMakerFeatureNestedStack extends FeatureNestedStack {
       featureName: `${featureName}`,
       featureCategory: 'media',
       updateCustomResourceProvider: props.updateCustomResourceProvider,
-      sageMakerInstanceType: 'ml.g4dn.xlarge',
+      sageMakerInstanceType: props.instanceType == null ? 'ml.g4dn.xlarge' : props.instanceType,
       sageMakerEcrDeployment: props.ecrDeployment,
     });
   }

diff --git a/src/api-deployment/features/sagemaker/general-nlu-sagemaker.ts b/src/api-deployment/features/sagemaker/general-nlu-sagemaker.ts
@@ -19,7 +19,7 @@ export class GeneralNLUSageMakerFeatureNestedStack extends FeatureNestedStack {
       featureName: `${featureName}`,
       featureCategory: 'media',
       updateCustomResourceProvider: props.updateCustomResourceProvider,
-      sageMakerInstanceType: 'ml.g4dn.xlarge',
+      sageMakerInstanceType: props.instanceType == null ? 'ml.g4dn.xlarge' : props.instanceType,
       sageMakerEcrDeployment: props.ecrDeployment,
     });
   }