diff --git a/mlops-multi-account-cdk/mlops-sm-project-template/mlops_sm_project_template/templates/finetune_deploy_llm_product/seed_code/build_app/setup.py b/mlops-multi-account-cdk/mlops-sm-project-template/mlops_sm_project_template/templates/finetune_deploy_llm_product/seed_code/build_app/setup.py index 8b1f619..fd5e228 100644 --- a/mlops-multi-account-cdk/mlops-sm-project-template/mlops_sm_project_template/templates/finetune_deploy_llm_product/seed_code/build_app/setup.py +++ b/mlops-multi-account-cdk/mlops-sm-project-template/mlops_sm_project_template/templates/finetune_deploy_llm_product/seed_code/build_app/setup.py @@ -34,7 +34,7 @@ "scipy==1.9.3", "diffusers==0.11.1", "datasets==2.8.0", - "transformers==4.33.0", + "transformers==4.38.0", "torch>=2.0.0", "peft==0.4.0", "accelerate==0.21.0", diff --git a/mlops-multi-account-cdk/mlops-sm-project-template/mlops_sm_project_template/templates/finetune_deploy_llm_product/seed_code/build_app/source_scripts/requirements.txt b/mlops-multi-account-cdk/mlops-sm-project-template/mlops_sm_project_template/templates/finetune_deploy_llm_product/seed_code/build_app/source_scripts/requirements.txt index bac7b0c..5004c23 100644 --- a/mlops-multi-account-cdk/mlops-sm-project-template/mlops_sm_project_template/templates/finetune_deploy_llm_product/seed_code/build_app/source_scripts/requirements.txt +++ b/mlops-multi-account-cdk/mlops-sm-project-template/mlops_sm_project_template/templates/finetune_deploy_llm_product/seed_code/build_app/source_scripts/requirements.txt @@ -1,4 +1,4 @@ -transformers==4.33.0 +transformers==4.38.0 torch>=2.0.0 peft==0.4.0 accelerate==0.21.0