diff --git a/official/projects/waste_identification_ml/Deploy/pet_grading_cloud_deployment/requirements.txt b/official/projects/waste_identification_ml/Deploy/pet_grading_cloud_deployment/requirements.txt
index 58d27b35852..fe46768a7ea 100644
--- a/official/projects/waste_identification_ml/Deploy/pet_grading_cloud_deployment/requirements.txt
+++ b/official/projects/waste_identification_ml/Deploy/pet_grading_cloud_deployment/requirements.txt
@@ -21,7 +21,7 @@ seaborn
 supervision
 matplotlib
 termcolor
-torch==2.6.0+cu124
+torch==2.12.0
 torchvision==0.21.0+cu124
 torchmetrics
 trackers
\ No newline at end of file