diff --git a/v3-examples/model-customization-examples/sft_finetuning_serverful_smtj.ipynb b/v3-examples/model-customization-examples/sft_finetuning_serverful_smtj.ipynb index f130931e54..fec96aaea9 100644 --- a/v3-examples/model-customization-examples/sft_finetuning_serverful_smtj.ipynb +++ b/v3-examples/model-customization-examples/sft_finetuning_serverful_smtj.ipynb @@ -88,7 +88,7 @@ " s3_output_path=S3_OUTPUT_PATH,\n", " compute=TrainingJobCompute(\n", " instance_type=\"ml.p5.48xlarge\",\n", - " instance_count=2,\n", + " instance_count=4,\n", " ),\n", " role=ROLE_ARN,\n", " base_job_name=\"sft-smtj\",\n", @@ -152,7 +152,7 @@ " s3_output_path=S3_OUTPUT_PATH,\n", " compute=TrainingJobCompute(\n", " instance_type=\"ml.p5.48xlarge\",\n", - " instance_count=2,\n", + " instance_count=4,\n", " ),\n", " role=ROLE_ARN,\n", " recipe=\"my_sft_recipe.yaml\",\n",